350. Intersection of Two Arrays II

Problem:

Solution:

在“349. Intersection of Two Arrays”问题的基础上稍作变动，349中不会显示重复元素，此题所有重复的交集元素都必须显示。只需先对两数组进行排序，剩下的工作就简单很多了，可以在O(n)时间内完成。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        int len1 = nums1.size();
        int len2 = nums2.size();
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        int i = 0;
        int j = 0;
        while (i < len1 && j < len2)
        {
            if (nums1[i] == nums2[j])
            {
                res.push_back(nums1[i]);
                i++;
                j++;
            }
            else if (nums1[i] < nums2[j]) i++;
            else j++;
        }
        return res;
    }
};