LeetCode 350. 两个数组的交集 II (哈希计数 / 排序+双指针)

 

思路： 1. 哈希表：利用哈希表对每个数组中的元素进行统计，并取最小个数即可。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        map<int, int> mp1, mp2;  // 开辟哈希表计数
        int len1 = nums1.size();
        int len2 = nums2.size();
        for(int i = 0; i < len1; i ++){  // 统计nums1中元素个数
            mp1[nums1[i]] ++;
        }
        for(int i = 0; i < len2; i ++){  // 统计nums2中元素个数
            mp2[nums2[i]] ++;
        }
        for(int i = 0; i < len2; i ++){
            if(mp1[nums2[i]]&&mp2[nums2[i]]){
                int cnt = min(mp1[nums2[i]], mp2[nums2[i]]);
                while(cnt --) ans.push_back(nums2[i]);  // 获取目标数组
                mp2[nums2[i]] = 0;
            }
        }
        return ans;
    }
};

2. 哈希表优化：上面的做法是进行了两次数量统计最后还要挨个push_back入答案，最坏的情况下的时间复杂度可能达到O(n+2m)或者O(2n+m)。而实际上我们可以只进行一次数量统计，在对第二个数组进行遍历时即可边遍历边push_back，时间复杂度为O(n+m)， 空间复杂度O(min(n, m))。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> ans;
        map<int, int> mp1;  // 开辟哈希表计数
        int len1 = nums1.size();
        int len2 = nums2.size();
        for(int i = 0; i < len1; i ++){  // 统计nums1中的元素
            mp1[nums1[i]] ++;
        }
        for(int i = 0; i < len2; i ++){  // 得到目标数组
            if(mp1[nums2[i]]){
                ans.push_back(nums2[i]);
                mp1[nums2[i]] --;
            }
        }
        return ans;
    }
};

3. 排序+双指针：如若两个数组都为有序的，那么就可以直接利用双指针进行交集的获取。因此我们可以先对数组进行排序，在利用两个指针扫描两个数组。初始情况下两个指针都位于数组头部，如若两指针对应元素相等push_back进答案，并两个指针都向后移动一位；如若不相等，则较小元素的指针向后移动一位，直至某一指针越界为止；时间复杂度O(nlogn+mlogm)，空间复杂度O(min(n, m))。

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());
        int len1 = nums1.size(), len2 = nums2.size();
        int i = 0, j = 0;
        while(i<len1&&j<len2){  // 双指针扫码
            if(nums1[i]==nums2[j]){
                ans.push_back(nums1[i]);
                i ++;
                j ++;
            }
            else if(nums1[i]<nums2[j]) i ++;
            else j ++;
        }
        return ans;
    }
};

4. 如若nums2存在磁盘中，无法一次完全读取完，那么此时排序+双指针的方法便不再可行，而需要利用哈希表进行计数和直接遍历nums2的方法即可。