leetcode 350. 两个数组的交集 II

问题描述

给定两个数组,编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:

输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:

如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

代码

稍微修改一下问题349. 两个数组的交集中的第一个代码:

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        map<int,int> table;
        int m = nums1.size(),n = nums2.size(),i;
        vector<int> ans;
        for(i = 0; i < m; i++)
        {
            table[nums1[i]]++;
        }
        for(i = 0; i < n; i++)
        {
            if(table[nums2[i]]!=0)
            {
                ans.push_back(nums2[i]);
                --table[nums2[i]];
            }
        }
        return ans;
    }
};

结果:

执行用时 :24 ms, 在所有 C++ 提交中击败了7.65%的用户
内存消耗 :10.8 MB, 在所有 C++ 提交中击败了5.16%的用户

代码

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

结果:

执行用时 :8 ms, 在所有 C++ 提交中击败了79.43%的用户
内存消耗 :10.4 MB, 在所有 C++ 提交中击败了5.16%的用户
posted @ 2020-04-02 08:10  曲径通霄  阅读(124)  评论(0编辑  收藏  举报