349. 两个数组的交集

349. 两个数组的交集

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

 方法1：排序查找o(n^2)

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        vector<int> temp;

        for(int i = 0;i<nums1.size();i++){
            for(int j = 0;j<nums2.size();j++){
                if(nums1[i] == nums2[j] && (!temp.size() || nums1[i] != temp.back())){
              //首先要相同，其次容器中元数的个数不为0或者最后一个与刚要加进去的数不相同，就加进去
                    temp.push_back(nums1[i]);
                }
            }
        }

        return temp;
    }
};

方法2：哈希表o(n)

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        //哈希表
        unordered_set<int> set1(nums1.begin(),nums1.end());
        unordered_set<int> set2;//保存结果

        for(int i = 0;i<nums2.size();i++){
            //发现nums2的元素 在set2里又出现过
            if(set1.find(nums2[i]) != set1.end()){
                set2.insert(nums2[i]);
            }
        }
        return vector<int>(set2.begin(),set2.end());//强行转换
    }
};