349两个数组的交集（双指针）

1、题目描述

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

说明：

• 输出结果中的每个元素一定是唯一的。
• 我们可以不考虑输出结果的顺序。

2、示例

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]

3、题解

基本思想：双指针，与algo里面的set_intersection算法一致

#include<iostream>
#include<vector>
#include<algorithm>
#include<set>
using namespace std;
class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        //基本思想：双指针，与algo里面的set_intersection算法一致
        vector<int> res;
        set<int> s1,s2;
        for(auto num:nums1)
            s1.insert(num);
        for(auto num:nums2)
            s2.insert(num);
        set<int>::iterator iter1=s1.begin(),iter2=s2.begin();
        while(iter1!=s1.end()&&iter2!=s2.end())
        {
            if(*iter1==*iter2)
            {
                res.push_back(*iter1);
                iter1++;
                iter2++;
            }
            else if(*iter1>*iter2)
                iter2++;
            else
                iter1++;
        }
        return res;
    }
};
int main()
{
    Solution solute;
    vector<int> nums1{1,2,2,1};
    vector<int> nums2{2,2};
    vector<int> nums=solute.intersection(nums1,nums2);
    for_each(nums.begin(),nums.end(),[](int n){cout<<n<<endl;});
    return 0;
}