LeetCode刷题|350两个数组的交集

本题来自于官方编辑的初级算法中数组分类下的一个题目。先看一下题目描述。

题目描述

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）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

前面两行刚开始我还以为是个递归，仔细一看原来只是为了找出比较短的数组。又学到了一个小技巧。