【LeetCode】C++ ：简单题 - 哈希表 350. 两个数组的交集 II

350. 两个数组的交集 II

难度简单435

给定两个数组，编写一个函数来计算它们的交集。

 

示例 1：

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

示例 2:

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

 

说明：

• 输出结果中每个元素出现的次数，应与元素在两个数组中出现次数的最小值一致。
• 我们可以不考虑输出结果的顺序。

进阶：

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

题解思路很清晰，启发了我：

用哈希表去存储数组元素及元素出现的次数，然后对比另一个数组，如果这个数组元素出现在哈希表中，那么将它加入最终结果数组里。

有几点需要注意：

题解中，在一开始进行数组长度的判断，这样的好处：

1. 不需要对数组进行额外判断，减少麻烦，可以直接以一种情况进行（本题中默认长的数组为nums1）。

2. 当情况时另一种时只需要将重新进入这个函数并将参数调换即可。

在以后写算法中，这样用参数调换位置就可以解决的办法实在是太棒了！

另外，当哈希表中该元素的个数为0时应该将其抹去，这样在后面的比对中才不会将数组1中不再多次出现的元素加入进去。这是个细节要考虑好，我一开始就没有想到。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        if(nums1.size() < nums2.size()){
            intersect(nums2, nums1);
        }

        unordered_map<int, int> map;
        for(auto& num: nums1){
            map[num]++;
        }
        vector<int> res;
        for(auto& num: nums2){
            if(map.count(num)){
                map[num]--;
                res.push_back(num);
                if(map[num] == 0){
                    map.erase(num);
                }
            }
        }
        return res;
    }
};

学习积累！加油！

 

---------------------复习更新---------------------

这里用map存储数字及其出现次数，因为对相同的数字是不会记在一起的，不需要考虑同一个出现多次的数字输出的问题。

而不用set, 如果是set会出现问题了，这时候需要每次额外判断字符出现的次数很不好做到。