LeetCode 350.两个数组的交集 ||

题目描述

给你两个整数数组 nums1 和 nums2 ，请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数，应与元素在两个数组中都出现的次数一致（如果出现次数不一致，则考虑取较小值）。可以不考虑输出结果的顺序。

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2,2]

示例 2:

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[4,9]

提示：

• 1 <= nums1.length, nums2.length <= 1000
• 0 <= nums1[i], nums2[i] <= 1000

进阶：

• 如果给定的数组已经排好序呢？你将如何优化你的算法？
• 如果 nums1 的大小比 nums2 小，哪种方法更优？
• 如果 nums2 的元素存储在磁盘上，内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

思路

由于题目要求把次数也输出，所以需要使用unordered_map来记录第一个数组中元素出现的次数，而不是使用unordered_set。遍历完第一个数组后，对于第二个数组中的每个数字，如果在哈希表中存在这个数字，则将该数字添加到答案，并减少哈希表中该数字出现的次数。

代码

C++版：

错误解法：

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> result;
        unordered_set<int> nums1_set(nums1.begin(),nums1.end());
        unordered_set<int> nums2_set(nums2.begin(),nums2.end());
        // 需要先对比两个数组的长度，先将长的放入哈希表里
        if(nums1.size()<nums2.size() || nums1_set.size()>nums2_set.size()){
            vector<int> tmp=nums1;
            nums1=nums2;
            nums2=tmp;
        }
        unordered_set<int> nums(nums1.begin(),nums1.end());
        for(int i=0;i<nums2.size();i++){
            if(nums.find(nums2[i])!=nums.end()){
                result.push_back(nums2[i]);
            }
        }
        return result;
    }
};

正确解法：

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> result;
        unordered_map<int,int> n;
        // 先交换，方便对较短的数组进行哈希表操作
        if(nums1.size()>nums2.size()){
            vector<int> tmp=nums1;
            nums1=nums2;
            nums2=tmp;
        }
        
        for(int i=0;i<nums1.size();i++){
            // 记录每个元素出现的次数
            n[nums1[i]]++;
        }
        for(int i=0;i<nums2.size();i++){
            if(n.count(nums2[i])){
                result.push_back(nums2[i]);
                n[nums2[i]]--;
                if(n[nums2[i]]==0){
                    n.erase(nums2[i]);
                }
            }
        }
        return result;
    }
};

Python版：

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        from collections import Counter
        result=list()
        if len(nums1) > len(nums2):
            tmp=nums1
            nums1=nums2
            nums2=tmp
        n=Counter()
        for num in nums1:
            n[num]+=1
        for num in nums2:
            count=n.get(num,0)
            if count>0:
                result.append(num)
                n[num]-=1
                if n[num]==0:
                    n.pop(num)
        
        return result

需要注意的地方

1.如果两个数组是有序的，则可以使用双指针的方法得到两个数组的交集。首先对两个数组进行排序，然后使用两个指针遍历两个数组。初始时，两个指针分别指向两个数组的头部。每次比较两个指针指向的两个数组中的数字，如果两个数字不相等，则将指向较小数字的指针右移一位，如果两个数字相等，将该数字添加到答案，并将两个指针都右移一位。当至少有一个指针超出数组范围时，遍历结束。