LeetCode | 两个数组的交集 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. 在同一数组上，找到了就删除元素；
2. 排序后，比较最后一个元素将不等于的元素删除即可；

三、程序实例

class Solution {
public:
    
    // 1. 在同一数组上，找到了就删除元素
    vector<int> intersect1(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ret;
        vector<int> minV = nums1.size() <= nums2.size() ? nums1 : nums2;
        vector<int> maxV = nums1.size() > nums2.size() ? nums1 : nums2;
        
        for(vector<int>::iterator minIter = minV.begin(); minIter != minV.end(); ++minIter)
        {
            for(vector<int>::iterator maxIter = maxV.begin(); maxIter != maxV.end(); ++maxIter)
            {
                if(*minIter == *maxIter)
                {
                    ret.push_back(*minIter);
                    minV.erase(minIter);
                    maxV.erase(maxIter);
                    minIter--;
                    break;
                }  
            }
        }
        return ret;
    }
    
    // 2. 排序后，比较最后一个元素将不等于的元素删除即可
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ret;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        
        while(!nums1.empty() && !nums2.empty())
        {
            if(nums1.back() > nums2.back())
                nums1.pop_back();
            else if(nums1.back() < nums2.back())
                nums2.pop_back();
            else
            {
                ret.push_back(nums1.back());
                nums1.pop_back();
                nums2.pop_back();
            }
        }
        return ret;
    }
};