一起玩转LeetCode 349. 两个数组的交集

95 阅读1分钟

349. 两个数组的交集

1、使用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());  
    }  
};