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

题目

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

例如:
给定 nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].

注意：

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

跟进:

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

思路一：

先排序两个数组，然后从第一个数组的每个元素遍历第二个数组，当第一第二个数组有相等的值，将值添入新数组，并从原数组中各自删除该值，直到两个数组遍历完。

代码：

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        for(int i=0;i<nums1.size();i++)
            for(int j=0;j<nums2.size();j++)
                if(nums1[i] == nums2[j]){
                    ans.push_back(nums1[i]);
                    nums1.erase(nums1.begin() + i);
                    nums2.erase(nums2.begin() + j);
                    i--;
                    break;
                }
        return ans;
    }
};

思路二（）优化：

既然排序了，可以利用两个数组大小关系，当第一个数组当前值大，第二个数组索引加 1；当第二个数组当前值大，第一个数组索引加 1；当一样大，将当前值加入答案数组，两个索引都加 1

代码：

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans = {};
        if(nums1.size() ==0 || nums2.size() == 0)
            return ans;
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        int n1 = 0, n2 = 0;
        while(n1 != nums1.size() && n2 != nums2.size()){
            if(nums1[n1] > nums2[n2])       n2++;
            else if(nums1[n1] < nums2[n2])  n1++;
            else{
                ans.push_back(nums1[n1]);
                n1++;
                n2++;
            }
        }
        return ans;
    }
};