LeetCode - Intersections of Two Arrays

解法一:

set insert -- O(log n)

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        set<int> temp(nums1.begin(), nums1.end()), res;
        for(int a: nums2){
            if(temp.find(a)!=temp.end())
                res.insert(res.end(), a);
        }
        return vector<int>{res.begin(), res.end()};
    }
};

解法二:

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int p1 = 0;
        int p2 = 0;
        while(p1<nums1.size() && p2<nums2.size()){
            if(nums1[p1]<nums2[p2]) p1++;
            else if(nums1[p1]>nums2[p2]) p2++;
            else{
                while(p1+1<nums1.size() && nums1[p1]==nums1[p1+1]) p1++;
                while(p2+1<nums2.size() && nums2[p2]==nums2[p2+1]) p2++;
                res.push_back(nums1[p1]);
                p1++;
                p2++;
            }
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/real_lisa/article/details/82756040