给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
是不是一看到求交集,就想调用函数,一行代码出结果?哈哈哈
这样就失去刷题的意义啦!因为笔试面试过程中你遇到相同的题概率真的特别小,你能做的是学会这种做题的思路,融会贯通,同时也要提升代码能力。
这题可以运用双指针p和q。先将两个数组排序,若指向的值相同,则添加到新的数组中, p 和 q 向右各移一位;若不相同,则将指向值更小的向右各移一位。
class Solution:
def intersect(self, nums1, nums2):
sortedNums1 = sorted(nums1)
sortedNums2 = sorted(nums2)
# print(sortedNums1,sortedNums2)
p = 0
q = 0
resultList = []
while(p < len(nums1) and q < len(nums2)):
if(sortedNums1[p] == sortedNums2[q]):
resultList.append(sortedNums1[p])
p += 1
q += 1
else:
if(sortedNums1[p] < sortedNums2[q]):
p += 1
else:
q += 1
return resultList