leetcode第349题两个数组的交集C++

原创于 2020-07-04 22:34:21 发布 · 177 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

81 篇文章

订阅专栏

本文介绍了一种计算两个数组交集的方法，通过使用C++的set容器，首先将第一个数组的元素插入到set中，然后遍历第二个数组，检查其元素是否在set中存在。如果存在，则将该元素添加到结果向量中，并从set中移除，以避免重复。

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

题目描述：给定两个数组，编写一个函数来计算它们的交集。

解题思路：先将nums1元素插入set容器再和nums2进行比较。

代码：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int>result;
        set<int>s1;
        for(int i=0;i<nums1.size();i++){
            s1.insert(nums1[i]);
        }
        for(int i=0;i<nums2.size();i++){
            if(s1.find(nums2[i])!=s1.end()){ 
            result.push_back(nums2[i]);
            s1.erase(nums2[i]);
        }
    }
        return result;
    }
};