LeetCode349 两个数组的交集

题目

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

示例

1. 示例 1:
   输入: nums1 = [1,2,2,1], nums2 = [2,2]
   输出: [2]
2. 示例 2:
   输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
   输出: [9,4]
   说明:
   出结果中的每个元素一定是唯一的。
   我们可以不考虑输出结果的顺序。

C++代码

不考虑输出结果，且输出结果不存在重复元素，采用set。
先将两个数组中重复元素去掉，然后进行比较。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) 
    {
        typedef set<int> numset;
        numset n1(nums1.begin(),nums1.end());
        numset n2(nums2.begin(),nums2.end());
        vector<int> inter;
        if(n1.size()<=n2.size())
        {
            for(numset::iterator iter=n1.begin();iter!=n1.end();iter++)
            {
                if(n2.find(*iter)!=n2.end()) inter.push_back(*iter);
            }
        }
        else
        {
            for(numset::iterator iter=n2.begin();iter!=n2.end();iter++)
            {
                if(n1.find(*iter)!=n1.end()) inter.push_back(*iter);
            }            
        }
        return inter;
            
    }
};