leetcode349 两个数组的交集

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

示例 1:

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

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

python3

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        s=set(nums1)
        return list(s.intersection(nums2))

C++

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