LeetCode 349. 两个数组的交集

LeetCode 349. 两个数组的交集

1、题目

力扣题目链接：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

2、使用set

代码

class Solution {  
public:  
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {  
        // 声明一个无序集合result_set，用于存储两个向量的交集元素  
        unordered_set<int> result_set;  
        // 将nums1向量转换为一个无序集合nums_set，这样可以更快地进行查找操作。其中使用begin()和end()函数来获取向量的起始和结束迭代器，以便将它们复制到nums_set中  
        unordered_set<int> nums_set(nums1.begin(), nums1.end());  
  
        // 遍历nums2中的每个元素，设其值为num  
        for (int num : nums2) {  
            // 检查num是否在nums_set中存在，如果存在则返回一个迭代器，否则返回end()，即指向nums_set的“末尾”（实际上是无效位置）。find()函数在哈希表中查找给定的值，如果不存在则返回end()  
            if (nums_set.find(num) != nums_set.end()) {  
                // 如果num存在于nums_set中，则将其添加到result_set中  
                result_set.insert(num);  
            }  
        }  
  
        // 返回result_set中的所有元素，作为一个新的整数向量。使用vector的构造函数，它接受两个迭代器（即result_set的开始和结束迭代器），并将它们复制到新的vector中  
        return vector<int>(result_set.begin(), result_set.end());  
    }  
};

复杂度分析

• 时间复杂度: O(m + n)

• 空间复杂度: O(n)

3、使用数组

代码

class Solution {  
public:  
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {  
        // 声明一个无序集合result_set，用于存储两个向量的交集元素  
        unordered_set<int> result_set;  
        // 声明一个名为hash的数组，用于存储nums1中的元素及其对应的标记（存在或不存在）  
        int hash[1000] = {0};  
  
        // 遍历nums1中的每个元素，将元素的值作为索引，将该索引对应的标记设为1，表示该元素存在于nums1中  
        for (int num : nums1) {  
            hash[num] = 1;  
        }  
  
        // 遍历nums2中的每个元素，检查该元素是否存在于hash中，如果存在（即hash中对应索引的标记为1），则将该元素插入到result_set中  
        for (int num : nums2) {  
            if(hash[num] == 1) {  
                result_set.insert(num);  
            }  
        }  
          
        // 返回result_set中的所有元素，作为一个新的整数向量  
        return vector<int>(result_set.begin(), result_set.end());  
    }  
};

复杂度分析

• 时间复杂度: O(m + n)

• 空间复杂度: O(n)