python3 给定两个数组,编写一个函数来计算它们的交集。

from typing import List


class Solution:
	def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
		# 记录两个数组中每个数字出现的次数
		result1 = {}
		result2 = {}
		
		# 遍历两个数组,然后将其数字出现次数添加入字典中
		for i in nums1:
			if i not in result1.keys():
				result1[i] = 1

			else:
				result1[i] += 1

		for i in nums2:
			if i not in result2.keys():
				result2[i] = 1
			else:
				result2[i] += 1

		# 将两个字典的键取出
		k1 = [i for i in result1.keys()]
		k2 = [i for i in result2.keys()]
		# 求两个字典键的交集
		k = set(k1) & set(k2)

		# 结果列表
		r = []
		
		# 同一个键,代表两个列表中都有出现,然后取最小的值为出现的次数,然后用列表的extend方法添加入结果列表中
		for i in k:
			m = min(result1[i], result2[i])
			r.extend([i]*m)

		return r

if __name__ == "__main__":
	s = Solution()
	nums1 = [1,2,2,1]
	nums2 = [2,2]
	r1 = s.intersect(nums1, nums2)
	print(r1)

给定两个数组,编写一个函数来计算它们的交集。

感觉只要用字典,速度都不会慢,因为字典查找遍历都好快

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章