LeetCode350. 两个数组的交集 II

最新推荐文章于 2024-05-05 23:55:50 发布

当如磐石

最新推荐文章于 2024-05-05 23:55:50 发布

阅读量87

点赞数

分类专栏： LeetCode刷题

本文链接：https://blog.csdn.net/oykotxuyang/article/details/106745488

版权

LeetCode刷题专栏收录该内容

77 篇文章 0 订阅

订阅专栏

350. 两个数组的交集 II

class Solution {
public:
    //时间O(n + m), 空间O(n + m)    (可以对较小的vector进行哈希映射，进一步优化空间为O(min(n, m)))
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size();
        int len2 = nums2.size();
        if(len1 == 0 || len2 == 0) return vector<int>();
        vector<int> res;
        unordered_map<int, int> mp1;
        unordered_map<int, int> mp2;
        for(auto& val : nums1) {
            mp1[val]++;
        }
        for(auto& val : nums2) {
            mp2[val]++;
        }
        for(auto& val: mp1) {
            int num = min(val.second, mp2[val.first]);
            if(num) {
                res.insert(res.end(), num, val.first);
            }
        }
        return res;
    }
};

class Solution {
public:
    //时间O(nlonn + mlogm), 空间O(1)(n，m为vector的大小)
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        sort(begin(nums1), end(nums1));
        sort(begin(nums2), end(nums2));
        int i = 0, j = 0, k = 0;
        while (i < nums1.size() && j < nums2.size()) {
            if (nums1[i] < nums2[j]) {
                ++i;
            } else if (nums1[i] > nums2[j]) {
                ++j;
            } else {
                nums1[k++] = nums1[i++];
                ++j;
            }
        }
        return vector<int>(begin(nums1), begin(nums1) + k);
    }
};