力扣 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. 初始化 unordered_set：首先创建了一个名为 nums_set 的 unordered_set，并使用 nums1的开始和结束迭代器进行初始化。这个步骤自动去除了 nums1 中的重复元素。
2. 遍历和查找：接着，遍历 nums2 中的每个元素。对于 nums2 的每个元素 num，使用 find 方法在 nums_set中查找。find 方法检查当前元素是否存在于 nums1 的集合表示中。
3. 插入结果集：如果 find 方法返回的迭代器不是 nums_set.end()（表示 num 存在于 nums_set 中），则将num 插入到 result 集合中。result 集合存储最终的交集结果，并且由于它是一个集合，它也会自动去除任何重复的元素。

时间复杂度：主要时间花费在遍历 nums2 并在 nums_set 中进行查找操作，平均情况下，unordered_set 的查找操作是常数时间的。因此，时间复杂度大约是 O(n + m)，其中 n 是 nums1 的长度，m 是 nums2 的长度。
空间复杂度：使用了两个额外的 unordered_set，空间复杂度也是 O(n + m)。

完整代码

#include<iostream>
#include<vector>
#include<unordered_set>
using namespace std;

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> result;
        unordered_set<int> nums_set(nums1.begin(), nums1.end()); // 去重
        for(int num : nums2){
            if(nums_set.find(num) != nums_set.end()){
                result.insert(num);
            }
        }
        return vector<int>(result.begin(), result.end());
    }
};
 
int main()
{
    Solution s;
    vector<int> nums1 = {1, 2, 2, 1};
    vector<int> nums2 = {2, 2};
    vector<int> result = s.intersection(nums1, nums2);
    for(int i = 0 ; i < result.size(); i++){
        cout << result[i] << " ";
    }
    return 0;
}