本题来自于官方编辑的初级算法中数组分类下的一个题目。先看一下题目描述。
题目描述
LeetCode地址:两个数组的交集
最开始看到这个题目时,我的想法很简单,既然求交集,那就是元素在两个数组中都出现过,那么我直接遍历一个数组,看这个数组中的元素是不是在另一个数组中也出现过,不就找到交集了吗。
于是我试了一下这个方法,但是提交的时候出错了。原因在于没有看清题目的一个要求,输出结果中每个元素出现的次数,应与元素在两个数组中出现次数的最小值一致。当我们输入为[2,1,2,1,2]和[2,2,1]时,输出值为[2,1,2,1,2],与题目要求不符。于是我看了一眼评论区,有提到使用双指针的方法,我就按照这个方式实现了一遍。
双指针法
首先先将数组排序,然后将指针分别指向两个数组第一个元素,对两个数组进行遍历。遍历过程有三种情况:
nums1的元素等于nums2的元素,那么说明这个数是交集的元素之一,放入交集数组中。
nums1的元素小于nums2的元素,就将指向nums1的指针后移一位。
nums1的元素大于nums2的元素,就将指向nums2的指针后移一位。
class Solution:
def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
intersection = []
nums1.sort()
nums2.sort()
m,n = len(nums1),len(nums2)
i,j = 0,0
while(i<m and j<n):
if nums1[i] == nums2[j]:
intersection.append(nums1[i])
i = i+1
j = j+1
elif nums1[i] < nums2[j]:
i = i+1
else:
j = j+1
return intersection
这就是双指针方法,实现起来也比较简单。另外官方还有一种使用哈希表的方法。
哈希表方法
先将较短的数组的元素做一个计数,映射到哈希表中,哈希表中是每个元素的出现次数。接着遍历另一个数组,如果哈希表中存在这个数组中的元素,那么就把元素加入到结果数组中,然后哈希表值减1,当值为0时删除这个元素。
class Solution:
def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
if len(nums1) > len(nums2):
return self.intersect(nums2, nums1)
m = collections.Counter()
for num in nums1:
m[num] += 1
intersection = list()
for num in nums2:
if (count := m.get(num, 0)) > 0:
intersection.append(num)
m[num] -= 1
if m[num] == 0:
m.pop(num)
return intersection
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/solution/liang-ge-shu-zu-de-jiao-ji-ii-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
前面两行刚开始我还以为是个递归,仔细一看原来只是为了找出比较短的数组。又学到了一个小技巧。