描述
返回两个数组的交
- Each element in the result must be unique.
- The result can be in any order.
您在真实的面试中是否遇到过这个题?
样例
nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2]
.
挑战
可以用三种不同的方法实现吗?
简单,用哈希表做:
class Solution:
"""
@param: nums1: an integer array
@param: nums2: an integer array
@return: an integer array
"""
def intersection(self, nums1, nums2):
# write your code here
if(len(nums1)==0):
return []
result=set()
dic={}
for i in range(len(nums1)):
dic[nums1[i]]=0
for i in range(len(nums2)):
if(nums2[i] in dic):
result.add(nums2[i])
return list(result)
s = Solution()
print(s.intersection([1,2,4,5], [4, 5]))