leetcode Intersection of Two Arrays

最新推荐文章于 2022-01-12 12:32:18 发布

柚子的power

最新推荐文章于 2022-01-12 12:32:18 发布

阅读量178

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/Exaggeration08/article/details/85375606

leetcode 专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种求解两个数组交集的高效算法，利用unordered_set特性，避免重复元素，加速查找过程。通过实例演示了如何使用C++实现此算法，包括源码解析和优化方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

本题要求两个集合的交集。

解题思路：对于两个数组nums1和nums2，把nums1中的值都存到unordered_set中，unordered_set的特点，存入的值不重复的，无序的，并且没有key value的说法，也就是说里面只存了一个值而不是一对值。遍历nums2中的每个元素，使用unordered_set中的count方法查找nums2中指定元素在unordered_set set_nums1中的个数，如果个数大于0，则说明该指定元素在nums1中存在，则把该元素插入到一个vector中，并在set_nums1中删除该元素。

以上是评论区中大佬的思路

源码附上：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> m(nums1.begin(), nums1.end());
        vector<int> res;
        for (auto a : nums2)
            if (m.count(a)) {
                res.push_back(a);
                m.erase(a);
            }
        return res;
    }
};

后来我又改了改，把nums1和nums2都存在到unordered_set中，使用set可以避免遍历重复元素，运行速度应该会更快一些；

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> m(nums1.begin(),nums1.end());
        unordered_set<int> n(nums2.begin(),nums2.end());
        vector<int> res;
        unordered_set<int>::iterator it=n.begin();
        
        for(it=n.begin();it!=n.end();it++)
        {
            if(m.count(*it)>0)
            {
                res.push_back(*it);
                m.erase(*it);
            }
        }
        return res;
    }
};