LeetCode350. 两个数组的交集 II

/**
 *
给定两个数组，编写一个函数来计算它们的交集。
示例 1：

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */

#include<vector>
#include <set>
#include <iostream>

using namespace std;

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        vector<int> result;
//        set_intersection 用于求两个集合的交集，两个区间必须是有序的，所以上面排了下序
//        back_inserter用于在末尾添加元素，将集合从末尾添加到result中
        set_intersection(nums1.begin(), nums1.end(), nums2.begin(), nums2.end(), back_inserter(result));
        return result;
    }
};

int main(){
    Solution s;
    vector<int> vec1,vec2,vec3;
    vec1 = { 4,9,5 };
    vec2 = { 9,4,9,8,4 };
    vec3 = s.intersect(vec1,vec2);
    vector<int>::iterator it;
    for (it = vec3.begin(); it !=vec3.end() ; it++) {
        cout<<*it<<endl;
    }
    return 0;
}

总结：

//在使用之前必须对两个集合先排序

//总结 set_intersection()求交集/ set_union()求并集/ set_difference()求差集
//形式:set_intersection(nums1.begin(), nums1.end(), nums2.begin(), nums2.end(), inserter(res, res.begin()))