LeetCode 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

思路

本题需要判断某个集合的元素是否出现在另一个集合里（求交集），可以使用哈希法。先将nums1放入unordered_set中，再让nums2与unordered_set中的元素进行对比是否相同，相同则存到结果集合中。

代码

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()); // 使用迭代器进行初始化
        for(int i=0;i<nums2.size();i++){
            // find返回的是迭代器
            if(nums.find(nums2[i])!=nums.end()){ // 发现nums2的元素 在nums里又出现过
                res.insert(nums2[i]);
            }
        }
        vector<int> res_vec(res.begin(),res.end());
        return res_vec;

    }
};

Python版：

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        # 使用哈希表存储一个数组中的所有元素
        table = dict()
        for num in nums1:
            table[num] = table.get(num, 0) + 1
        # 使用集合存储结果
        res = set()
        for num in nums2:
            if num in table:
                res.add(num)
                del table[num]
        return list(res)

需要注意的地方

1.如果哈希值比较少、特别分散、跨度非常大，使用数组就造成空间的极大浪费。

2.由于std::set和std::multiset底层实现都是红黑树，std::unordered_set的底层实现是哈希表， 使用unordered_set 读写效率是最高的，并不需要对数据进行排序，而且还不要让数据重复，所以本题选择unordered_set。