LeetCode - Intersections of Two Arrays

最新推荐文章于 2019-03-10 17:53:03 发布

real_lisa

最新推荐文章于 2019-03-10 17:53:03 发布

阅读量108

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/real_lisa/article/details/82756040

LeetCode 专栏收录该内容

180 篇文章

订阅专栏

本文介绍三种高效查找两个整数数组交集的方法：使用集合进行查找、双指针技巧及集合交并操作。通过不同算法组合，实现快速查找并返回两个输入数组共有的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

两个array中有什么重复元素。

insert input arrays into set 跟 sorting 基本可以替换，但是set需要extra space

set_intersection 又可以和 two pointers 找相同元素可以替换

set search 也可以和 binary search 替换

选你喜欢的组合吧～

解法一

set insert / search -- O(log n); construct the whole set O(n logn)

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        set<int> temp(nums1.begin(), nums1.end()), res;
        for(int a: nums2){
            if(temp.find(a)!=temp.end())
                res.insert(res.end(), a);
        }
        return vector<int>{res.begin(), res.end()};
    }
};

解法二

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int p1 = 0;
        int p2 = 0;
        while(p1<nums1.size() && p2<nums2.size()){
            if(nums1[p1]<nums2[p2]) p1++;
            else if(nums1[p1]>nums2[p2]) p2++;
            else{
                while(p1+1<nums1.size() && nums1[p1]==nums1[p1+1]) p1++;
                while(p2+1<nums2.size() && nums2[p2]==nums2[p2+1]) p2++;
                res.push_back(nums1[p1]);
                p1++;
                p2++;
            }
        }
        return res;
    }
};

解法三

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        set<int> s1(nums1.begin(), nums1.end()), s2(nums2.begin(), nums2.end()), res;
        set_intersection(s1.begin(), s1.end(), s2.begin(), s2.end(), inserter(res, res.begin()));
        return vector<int>(res.begin(), res.end());
    }
};