leetcode 350. 两个数组的交集 II

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

说明：

输出结果中每个元素出现的次数，应与元素在两个数组中出现次数的最小值一致。

我们可以不考虑输出结果的顺序。

进阶：

如果给定的数组已经排好序呢？你将如何优化你的算法？

如果 nums1 的大小比 nums2 小很多，哪种方法更优？

如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，怎么办？

 

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2]

输出：[2,2]

示例 2:

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]

输出：[4,9]

 

解题思路：

• 先对两数组排序，然后指针逐个走

• 与另一数组相等则加到结果里

• 当1里值比2里值小时，则需要在1里往后走找一个大数

• 当1里值比2里值大时，需要在2里往后走找一个大数

C/C++题解：

class Solution {

public:

    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {

        vector<int> res;

        int m = nums1.size(), n = nums2.size();

        if(m == 0 || n == 0) return res; //有空数组交集为空

        if(m > n)  return intersect(nums2, nums1);//以短数组为基准遍历

        sort(nums1.begin(), nums1.end());//从小到大排序

        sort(nums2.begin(), nums2.end());

        int i=0, j=0;//经过排序比较分三种

        while(i < m && j < n){//遍历短数组

            if(nums1[i] == nums2[j]){//与另一数组相等则加到结果里

                res.push_back(nums1[i]);

                i++;

                j++;}//当1里值比2里值小时，则需要在1里往后走找一个大数

            else if(nums1[i] < nums2[j]){ i++;

            }//当1里值比2里值大时，需要在2里往后走找一个大数

            else j++;}

        return res;}};

Debug结果：

Java题解：

class Solution {

    public int[] intersect(int[] nums1, int[] nums2) {

        int m = nums1.length, n = nums2.length;

        if(m == 0 || n == 0) return new int[0]; //有空数组交集为空

        if(m > n)  return intersect(nums2, nums1);//以短数组为基准遍历

        int[] res = new int[m];

        Arrays.sort(nums1);//从小到大排序

        Arrays.sort(nums2);

        int i=0, j=0, k=0;//经过排序比较分三种

        while(i < m && j < n){//遍历短数组

            if(nums1[i] == nums2[j]){//与另一数组相等则加到结果里

                res[k++] = nums1[i];

                i++;

                j++;}//当1里值比2里值小时，则需要在1里往后走找一个大数

            else if(nums1[i] < nums2[j]){ i++;

            }//当1里值比2里值大时，需要在2里往后走找一个大数

            else j++;}

        return res;}}

Debug结果：

Python题解：

class Solution(object):

    def intersect(self, nums1, nums2):

        """:type nums1: List[int]:type nums2: List[int]

        :rtype: List[int]"""

        res = []

        m, n = len(nums1), len(nums2)

        if m == 0 or n == 0: return res #有空数组交集为空

        if m > n:  return Solution.intersect(self, nums2, nums1) #短数组为基准遍历

        nums1.sort() #从小到大排序

        nums2.sort()

        i, j= 0, 0 #经过排序比较分三种

        while i < m and j < n: #遍历短数组

            if nums1[i] == nums2[j]: #与另一数组相等则加到结果里

                res.append(nums1[i])

                i, j = i+1, j+1

            #当1里值比2里值小时，则需要在1里往后走找一个大数

            elif nums1[i] < nums2[j]: i += 1

            #当1里值比2里值大时，需要在2里往后走找一个大数

            else: j += 1

        return res

Debug结果：

更多题解移步公众号免费获取