LeetCode349. 两个数组的交集

349. 两个数组的交集 - 力扣（LeetCode）

给定两个数组 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

题解

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> result_set;
        unordered_set<int> nums_set(nums1.begin(), nums1.end());
        //unordered_set 自动去重

        for (int num : nums2) {
            if (nums_set.find(num) != nums_set.end()) {
                result_set.insert(num);
            }
        }

        return vector<int>(result_set.begin(), result_set.end());
    }
};

解题思路

这道题求两个数组的交集，核心思路是先把第一个数组转成哈希集合实现去重，然后遍历第二个数组，检查每个元素是否在这个哈希集合中存在，如果存在就把它加入结果集合（自动实现去重），最后把结果集合转成数组返回。

展开解释一下核心逻辑

for (int num : nums2) {
    if (nums_set.find(num) != nums_set.end()) {
        result_set.insert(num);
    }
}

  for (int num : nums2)

• 遍历 nums2 vector 中的每一个元素
• 每次循环，当前元素会被赋值给变量 num
• 比如 nums2 = [2,2]，那么 num 第一次是 2，第二次还是 2

      nums_set.find(num)

• find() 是 unordered_set 的成员函数
• 它在集合中查找值为 num 的元素
• 如果找到了，返回指向该元素的迭代器
• 如果没找到，返回 nums_set.end()

  nums_set.find(num) != nums_set.end()

• 比较两个迭代器
• 如果 find() 返回的迭代器不等于 end()，说明找到了元素
• 如果等于 end()，说明没找到元素

     result_set.insert(num)

• 当找到匹配的元素时，将这个元素插入到结果集合中
• 因为使用的是 insert，所以即使插入重复元素，也只会保存一份