350. Intersection of Two Arrays II

时间：2017-02-11 15:50:40 阅读：202 评论：0 收藏：0 [点我收藏+]

标签：data static float text should font 实现 als tar

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

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

Note:

Each element in the result should appear as many times as it shows in both arrays.
The result can be in any order.

Follow up:

What if the given array is already sorted? How would you optimize your algorithm?
What if nums1‘s size is small compared to nums2‘s size? Which algorithm is better?
What if elements of nums2 are stored on disk, and the memory is limited such that you cannot load all elements into the memory at once?
来源： https://leetcode.com/problems/intersection-of-two-arrays-ii/

分析

算法1

第一种方法，使用multiset，建立两个set，并且遍历较短的那个nums。这种方法适用于排序和未排序的数组

set的构造时间复杂度 N*log(N)

erase： log(N) + target.count 重复N次，所以erase总时间复杂度 N*log(N)

所以该算法时间复杂度为 N*log(N)

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        multiset<int> set1(nums1.begin(), nums1.end());
        multiset<int> set2(nums2.begin(), nums2.end());
        vector<int> result;
        vector<int> & nums = nums1.size() > nums2.size() ? nums2 : nums1;// nums equals to the shorter array
        for(auto n : nums){
            int a = set1.erase(n);// find the count of n in nums1
            int b = set2.erase(n);// find the count of n in nums2
            if( a && b){
                int time = min(a,b);
                while(time--)
                    result.push_back(n);            
            }
        }
        return result;
    }
};

算法2

可以使用系统函数 set_intersection求解两个已经排序好的数组的交集

http://www.cplusplus.com/reference/algorithm/set_intersection/

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> result(nums1.size() + nums2.size());
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        auto it = set_intersection(nums1.begin(), nums1.end(), nums2.begin(), nums2.end(), result.begin());
        result.resize(it - result.begin());
        return result;
    }
};

可以自己实现交集功能

比如下面的两个排序好的数组，用两个指针分别指向开头，诶个遍历，直到有一个遍历结束

如果 A[i] < B[j]，i++

否则如果 B[j] < A[i]，j++

否则（这里一定是二者相等）push 当前值到 result，并且i++,j++

时间复杂度是O(N+M)

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:1

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:1,1

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:1,1,2

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:1,1,2

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:1,1,2

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:1,1,2

A:1 1 2 3 3 4 4 4 4 5

B:0 0 1 1 2 2 4 4 6 result:1,1,2,4

A:1 1 2 3 3 4 4 4 4 5