leetcode349. 两个数组的交集

简单题，竟然想了20分钟= =

题目

给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。

示例

示例 1：

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

示例 2：

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]
解释：[4,9] 也是可通过的

提示：

• 1 <= nums1.length, nums2.length <= 1000
• 0 <= nums1[i], nums2[i] <= 1000

思路：

把两个vector分别放到set中（自动去重） ，然后遍历第一个set，如果在第二个set中可以找到该元素，就把它加入到 交集数组中。

代码：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        // 思路：两个数组去重，然后遍历nums1，查询nums2中是否有对应的值
        set<int> st1, st2;
        int i = 0;
        for(i = 0; i < nums1.size(); i++){
            st1.insert(nums1[i]);
        }
        for(i = 0; i < nums2.size(); i++){
            st2.insert(nums2[i]);
        }
        vector<int> ans;
        for(set<int>::iterator it = st1.begin(); it!=st1.end(); it++){
            if(st2.find(*it) != st2.end()){
                ans.push_back(*it);
            }
        }
        return ans;
    }
};

运行结果：

知识点：

set的遍历，用迭代器it

在 set 中查找值，用 set.find(value)，它返回的值是迭代器，如果没有找到就是 最后一个元素的后面 即 st.end()