LeetCode--349 两个数组的交集

题目

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

示例

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:

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

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        //经过 set 之后，重复的元素被删除 与运算对于集合来说就是求交集
        set<int>s1(nums1.begin(), nums1.end());
        set<int>s2(nums2.begin(), nums2.end());
        
        set<int>::iterator it1 = s1.begin();
        set<int>::iterator it2 = s2.begin();
        
        vector<int> res ;
        while(it1 != s1.end() && it2 != s2.end())
        {
            if(*it1 == *it2)
            {
                res.push_back(*it1);
                 it1++;
                 it2++;
            } else if(*it1 < *it2)
            {
                it1++;
            }
            else
            {
                it2++;
            }    
        }
        
        return res;
    }
};