349. Intersection of Two Arrays - Easy

Description

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.

Analyze

上次那道动态规划的题还没解出来，难受，就做道简单题寻求安慰。。。
排序下面找的，所以思路也非常明了，排序完了用两个指针。

Code

vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
    sort(nums1.begin(), nums1.end());
    sort(nums2.begin(), nums2.end());
    vector<int> ret;
    for(int i = 0, j = 0; i < nums1.size() && j < nums2.size();) {
        if(nums1[i] == nums2[j]) {
            if(ret.empty() || nums1[i] != ret.back()) ret.push_back(nums1[i]);
            i++;
            j++;
        }
        else if(nums1[i] < nums2[j]) ++i;
        else ++j;
    }
    return ret;
}

Discuss

评论区还有种用hash set的方法，第一步扫描第一个数组得到一个set，第二步扫描另一个数组，检查每个元素是否在上一个set中再放进set得到intersection。