LintCode-[容易] 548. 两数组的交II

描述:

计算两个数组的交

注意事项:

每个元素出现次数得和在数组里一样
答案可以以任意顺序给出

样例:

nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].

思路:

把538. 两数组的交中的删除重复元素的操作去掉即可。

C++实现:

class Solution {
public:
    /**
     * @param nums1 an integer array
     * @param nums2 an integer array
     * @return an integer array
     */
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        // Write your code here
        int size1 = nums1.size();
        int size2 = nums2.size();
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int i1 = 0;
        int i2 = 0;
        vector<int>result;
        while (i1 < size1 && i2 < size2) {
            if (nums1[i1] == nums2[i2]) {
                result.push_back(nums1[i1]);
                i1++;
                i2++;
            }
            else if (nums1[i1] < nums2[i2]) {
                i1++;
            }
            else {
                i2++;
            }
        }
        return result;
    }
};

你可能感兴趣的:(-LintCode-)