两个数组的交集--力扣349

题目

给定两个数组 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] 也是可通过的

思路

首先当需要在集合里面判断一个元素是否出现过，优先考虑哈希表。这里我们使用unordered_set，因为不需要排序且要去重。
我们定义两个unordered_set，一个用于装入num1的数据，另外一个用于保存交集。然后通过遍历nun2的数据在第一个unordered_set中找出现过的数据，如果没有找到nums.find(num) 返回的是迭代器nums.end()，找到就直接存储起来。
最后返回的是一个数组，所以使用vector转换一下。

C++代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> res;//用于存放交集
        unordered_set<int> nums(nums1.begin(), nums1.end());//放入num1数组的值并去重
        for(int num : nums2){
            //如果存在交集，则放入res中
            if(nums.find(num) != nums.end()) res.insert(num);
        }
        //结果要返回数组
        return vector<int>(res.begin(), res.end());
    }
};