[LeetCode] Intersection of Two Arrays 两个数组

 

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

Note:

• Each element in the result must be unique.
• The result can be in any order.

 

这道题让我们找两个数组相同的部分，难度不算大，我们可以用个set把nums1都放进去，然后遍历nums2的元素，如果在set中存在，说明是公共部分，加入结果的set中，最后再把结果转为vector的形式即可：

 

解法一：

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());
    }
};

 

或者我们也可以使用STL的set_intersection函数来找出共同元素，很方便：

 

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());
    }
};

 

参考资料：

https://leetcode.com/discuss/103295/my-c-solutions-using-set-and-unordered_set

 

LeetCode All in One 题目讲解汇总(持续更新中...)