leetcode349

题目：

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]

示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]

说明:

• 输出结果中的每个元素一定是唯一的。
• 我们可以不考虑输出结果的顺序。

 

解题思路：

声明两个set s和res；

把nums1中的元素放到s中去；

遍历nums2，如果nums2中的元素在s中，则把该元素放入res中去；

把res转化成vector返回。

 

代码：

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