LeetCode 349. 两个数组的交集

题目描述： 两个数组的交集

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

例子:

         给定 num1= [1, 2, 2, 1], nums2 = [2, 2], 返回 [2].

提示:

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

解题思路：

        交集问题，用集合set做肯定简单，把一个数组全插入一个集合中，在用第二个数组的数检查集合，如果集合有这个数的话把数压入返回的数组中，再把那个数从集合删除就行了。

代码：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        set<int> s;
        vector<int> v;
        for(int i = 0; i < nums1.size(); i ++) 
            s.insert(nums1[i]);
        for(int i = 0; i < nums2.size(); i ++){
            if(s.find(nums2[i]) != s.end()){
                v.push_back(nums2[i]);
                s.erase(nums2[i]);
            }
        }
        return v;
    }
};