给定两个数组 nums1
和 nums2
,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
例:
输入:nums1 = [1,2,2,1], nums2 = [2,2] 输出:[2]
解析:
对两个数组进行去重,然后进行升序排列,从小到大依次进行比较,相同得值放入结果数组,两个数组指针有一个到顶了就退出循环比较。最后返回结果数组即可。
class Solution(object):
def intersection(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
nums1 = list(set(nums1)) # 去重
nums1.sort() # 升序排列
nums2 = list(set(nums2))
nums2.sort()
i = 0 # 数组1指针
j = 0 # 数组2指针
res = [] # 结果数组
while i < len(nums1) and j < len(nums2): # 循环条件
if nums1[i] == nums2[j]: # 两个值想等得情况
res.append(nums1[i]) # 将值加入结果数组
i += 1 # 数组1指针后移
j += 1 # 数组2指针后移
elif nums1[i] < nums2[j]: # 数组1得值小于数组2
i += 1 # 数组1得指针后移
elif nums1[i] > nums2[j]: # 数组1得值大于数组2
j += 1 # 数组2得指针后移
return res # 返回结果数组