LeetCode-349 Intersection of Two Arrays

https://leetcode.com/problems/intersection-of-two-arrays/

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

Note:

• Each element in the result must be unique.
• The result can be in any order.

1、先排序，再用两指针分别扫描两数组（12ms）

class Solution {

public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> num;
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        int i = 0,j = 0;
        while(i<nums1.size() && j < nums2.size()){
            if(nums1[i] == nums2[j] && find(num.begin(),num.end(),nums1[i]) == num.end()){
                num.push_back(nums1[i]);
                i++;
                j++;
                continue;
            }
            if(nums1[i]<nums2[j])
                i++;
            else
                j++;
        }
    return num;
    }

};

2、发现STL中有set_intersection()函数，是对于排好序后的两个集合（即不会有重复元素）求交集，因此这里调用该函数的话，会有重复元素出现。

2.1利用unique、erase去重：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        vector<int> num(100000);   // 数组长度影响：100000——28ms，500——12ms
        auto iter = set_intersection(nums1.begin(),nums1.end(),nums2.begin(),nums2.end(),num.begin());
        num.resize(iter-num.begin());
        vector<int>::iterator iter1 = unique(num.begin(),num.end());
        num.erase(iter1,num.end());
    return num;
    }
};

2.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> num(1000),result;// 同上，数组长度影响：100000——28ms，（500、1000等）——12ms
        auto iter = set_intersection(nums1.begin(),nums1.end(),nums2.begin(),nums2.end(),num.begin());
        num.resize(iter-num.begin());
        int len = num.size();
        for(int i = 0;i<len;i++){
            if(! result.empty() && num[i]==result.back())
                continue;
            result.push_back(num[i]);
        }
    return result;
    }
};