题目描述:
给定两个数组 nums1
和 nums2
,返回 它们的 交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2] 输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出:[9,4] 解释:[4,9] 也是可通过的
提示:
1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000
解法一:卡python的bug直接使用set( )提取集合的交集,但是在其他语言中不能直接这么用。
class Solution:
def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
return list(set(nums1) & set(nums2))
解法二:哈希法
使用字典和集合实现哈希。使用集合存储其中一个数组,再判断另一个数组中的元素在不在此数组中,讲结果add到集合里。
class Solution:
def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
one = {}
for num in nums1:
one[num] = one.get(num,0) + 1
result = set()
for i in nums2:
if i in one:
result.add(i)
del one[i]
return list(result)
使用数组实现哈希,因为题中给了nums数组大小的范围,如果没给则不可用。两个表记录数字出现的次数,一个表记录结果。
class Solution:
def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
count1 = [0]*1001
count2 = [0]*1001
result = []
for i in range(len(nums1)):
count1[nums1[i]] += 1
for j in range(len(nums2)):
count2[nums2[j]] += 1
for k in range(1001):
if count1[k]*count2[k] > 0:
result.append(k)
return result