【leetcode】350. 两个数组的交集 II( Intersection of Two Arrays II )

---

题目描述

【leetcode】350. 两个数组的交集 II( Intersection of Two Arrays II )

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

示例 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，并添加该字符到结果。

代码：

#include <unordered_map>
class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        //默认nums1更短，若不是，我们通过递归的方式巧妙的交换元素
        if(nums1.size() > nums2.size()){
            return intersect(nums2, nums1);
        }

        // 第一次遍历，把长度较短的数组添加到哈希表，并计数每个数字出现次数。   
        unordered_map<int, int> count;
        for(int i=0; i<nums1.size(); ++i){
            if(count.end() == count.find(nums1[i])){
                count[nums1[i]] = 1;
            }
            else{
                ++count[nums1[i]];
            }
        }

        // 第二次遍历另一个数组，若与哈希表中重复，则哈希表次数-1，并添加该字符到结果。
        vector<int> result;
        for(int i=0; i<nums2.size(); ++i){
            if(count.end() != count.find(nums2[i])){
                --count[nums2[i]];
                result.push_back(nums2[i]);
                // 如果元素计数为0，删掉它
                if(0 == count[nums2[i]]){
                    count.erase(nums2[i]);
                }
            }
        }
        return result;
    }
};

结果：

相关/参考链接