LeetCode : 349. 两个数组的交集

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] 也是可通过的

提示：

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

解题思路：创建一个哈希表index来存放交集。遍历num1 把出现的数字存起来置为1，再遍历num2 把跟num1重合的数字置为0，遍历哈希表index，当值为0的关键字就是重叠的数字。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        map<int,int> index;
        for(int i = 0; i < nums1.size(); i++)
        {
            if(index.find(nums1[i]) == index.end())
                index[nums1[i]] = 1;
        }
        for(int i = 0; i < nums2.size(); i++)
        {
            if(index.find(nums2[i]) != index.end())
                index[nums2[i]] = 0;
            
        }
        map<int, int> ::iterator it;
        vector<int> res;
        for(it = index.begin(); it != index.end(); it++)
        {
            if(it->second == 0) 
                res.push_back(it->first);
        }
        return res;
    }
};

解题思路：使用unordered_set<int>来存放结果，首先先把num1的重复数字去掉，再遍历num2，当num1 和 num2 的数字相同时，把该数字加入到存放结果的res中，最后输出的时候需要把res中的重复数字去掉。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        //用unordered_set来存储唯一的结果
        unordered_set<int> res;
        unordered_set<int> index(nums1.begin(), nums1.end());//去掉num1的重复数字
        for(int num : nums2)
        {
            if(index.find(num) != index.end())//当num2中存在与num1一样的数字时，把结果放到res中
                res.insert(num);
        }
        return vector<int>( res.begin(),res.end());//去掉res中的重复数字
    }
};