Leetcode 349. Intersection of Two Arrays

349. Intersection of Two Arrays

• Total Accepted: 33720
• Total Submissions: 75833
• Difficulty: Easy

 

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.

 

思路：求交集。

 

代码：

方法一：直接用unordered_set，复杂度比下面的方法低。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> m(nums1.begin(), nums1.end());
        vector<int> res;
        for (int a:nums2)
            if (m.erase(a)) res.push_back(a);
        return res;
    }
};

 

方法二：先排序，再遍历，nums1[i]==nums2[j]时插入新集合res，注意重复元素的情况。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        vector<int> res;
        int i=0,j=0;
        while(i<nums1.size()&&j<nums2.size()){
            while(i+1<nums1.size()&&nums1[i]==nums1[i+1]) i++;//去除重复元素
            while(j+1<nums2.size()&&nums2[j]==nums2[j+1]) j++;
            if(nums1[i]==nums2[j]){
                res.push_back(nums2[j]);
                i++;
                j++;
            }
            else{
                nums1[i]>nums2[j]?j++:i++;
            }
            /*
            //else部分改成下面就是求并集
            else{
                nums1[i]>nums2[j]?res.push_back(nums2[j++]):res.push_back(nums1[i++]);
            }
            */
        }
        return res;
    }
};

 

方法三：先排序，再用二分查找，注意二分查找的返回条件。

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(!nums1.size()||!nums2.size()) return res;
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        for(int i=0;i<nums1.size();i++){
            while(i+1<nums1.size()&&nums1[i]==nums1[i+1]) i++;
            if(BinarySearch(nums2,nums1[i])){
                res.push_back(nums1[i]);
            }
        }
        return res;
    }
    bool BinarySearch(vector<int> nums2,int num){
        int left=0,right=nums2.size()-1,mid;
        while(left<=right){
            mid=left+(right-left)/2;
            if(nums2[mid]<num){
                left=mid+1;
            }
            else{
                right=mid-1;
            }
        }
        return left<nums2.size()&&nums2[left]==num;
    }
};