欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Binary Search:349. Intersection of Two Arrays

程序员文章站 2022-07-15 10:50:19
...

Binary Search:349. Intersection of Two Arrays

求两个数组相交的数字的集合,结果中不能有重复的。

想到了用set

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size(), n = nums2.size();
        set<int> res;
        vector<int> result;
        for(int i = 0; i < m; ++i)
        {
            for(int j = 0; j < n; ++j)
            {
                if(nums1[i] == nums2[j])
                {
                    res.insert(nums1[i]);
                    break;
                }
            }
        }
        for(auto a : res)
        {
            result.push_back(a);
        }
        return result;
    }
};

看了大神代码,更简洁:

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