题目
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
解题思路
一开始,我采取了最粗暴的方法,遍历一个列表,拿出每一个元素去判断该元素是否在另一个列表中,如果在,就把这个元素拿出来放在一个地方,同时在另一个列表中删除该元素。最后所有拿出来的元素就是交集。
显然,这很慢。
开始想第二种方法,把这两个列表都转成字典,值是每个元素出现的次数,然后再来遍历某个字典,依次拿出key去判断是否在另一个字段中,如果存在,再取出2个字典中,值小的那个一个并乘以值作为个数,用一个列表接收。
这个,比上面快了一点点。
别人提交的最快的答案:
其实我的第二种思路已经接近了,但是没有必要将两个列表都转换成字典,只要将一个列表转换成字典,然后遍历另一个列表,这样就将复杂度降低了一个N。
答案(一)
class Solution(object):
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
result = []
for i in nums1:
if i in nums2:
nums2.remove(i)
result.append(i)
return result
答案(二)
from collections import Counter
class Solution(object):
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
result = []
nums1_dict = Counter(nums1)
nums2_dict = Counter(nums2)
for i in nums1_dict:
if i in nums2_dict:
result += [i] * min(nums1_dict[i], nums2_dict[i])
return result
答案(三)
class Solution(object):
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
dictnums1 = dict()
ans = []
for num in nums1:
dictnums1[num] = dictnums1[num] + 1 if num in dictnums1 else 1
for num in nums2:
if num in dictnums1 and dictnums1[num]>=1:
ans.append(num)
dictnums1[num] -=1