LeetCode350 两个数组j的交集II

题目

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

示例

1. 示例 1:
   输入: nums1 = [1,2,2,1], nums2 = [2,2]
   输出: [2,2]
2. 示例 2:
   输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
   输出: [4,9]
   说明：
   1.输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
   2.我们可以不考虑输出结果的顺序。
   进阶:
   1.如果给定的数组已经排好序呢？你将如何优化你的算法？
   2.如果 nums1 的大小比 nums2 小很多，哪种方法更优？
   3.如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

C++代码

一开始的想法是不用map，创建三个vector：lnum（存放较长的数组）、snum（存放较短的数组）和inter（存放结果）。
从snum的首位开始遍历，如果元素出现在lnum中，则存入inter中。
但是这种方法在两个数组等长时会出错，比如[1,1]和[1,2],最终会返回[1,1]而不是[1]。

然后改为了用map；mm用于存放nums1中元素（键）和其出现的次数（值），然后遍历nums2，如果该元素为mm的键，且其值不为0，则将其存入inter中，然后将键对应的值减1.

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) 
    {
        map<int,int> mm;
        vector<int> inter;
        for(int i=0;i<nums1.size();i++)
            mm[nums1[i]]++;
        for(int i=0;i<nums2.size();i++)
        {
            if(mm.find(nums2[i])!=mm.end()&&mm[nums2[i]]>0) 
            {
               inter.push_back(nums2[i]);
                mm[nums2[i]]--;
            }
        }
        return inter;
    }
}；