【刷题训练】349. 两个数组的交集（set的使用）

最新推荐文章于 2024-06-12 10:00:23 发布

阿然成长日记

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量203

点赞数 2

分类专栏：数据结构与算法文章标签：算法 leetcode

本文链接：https://blog.csdn.net/luhaoran814/article/details/137522121

版权

数据结构与算法专栏收录该内容

33 篇文章 1 订阅

订阅专栏

349. 两个数组的交集

在这里插入图片描述

C++:

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        //先将两个数组放入set中，去重并且排序。
        set<int> set1(nums1.begin(),nums1.end());
        set<int> set2(nums2.begin(),nums2.end());
        vector<int> ret;
        //取起始位置
        auto it1 = set1.begin();
        auto it2 = set2.begin();

        while(it1!=set1.end()&&it2!=set2.end())
        {
            if(*it1<*it2)
            {
                it1++;
            }
            else if(*it1>*it2)
            {
                it2++;
            }
            else
            {
                ret.push_back(*it1);
                it1++; 
                it2++;       
            }
        }
        return ret;
    }
};