LeetCode 350. 两个数组的交集 II(java)

柔光的暖阳◎ 2022-04-04 08:19 224阅读 0赞

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

**示例 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 *的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

思路：

先将两个数组排序，再对俩数组分别设置遍历指针，进行循环判断，如：   
nums1 = \[1,3,4,7,20,35\] nums2 = \[4,15,35\]   
i 遍历 nums1，j 遍历 nums2，初值都为 0   
nums1\[i\]=1 < nums2\[j\]=4 i++   
nums1\[i\]=3 < nums2\[j\]=4 i++   
nums1\[i\]=4 = nums2\[j\]=4 ，即 4 是交集，i++,j++   
nums1\[i\]=7 < nums2\[j\]=15 i++   
nums1\[i\]=20 > nums2\[j\]=15 j++

class Solution {
        public int[] intersect(int[] nums1, int[] nums2) {
            int m = nums1.length;
            int n = nums2.length;
            if (m == 0 || n == 0)
                return new int[] {};
            int[] ans = new int[m + n];
            Arrays.sort(nums1);
            Arrays.sort(nums2);
            int i = 0;
            int j = 0;
            int k = 0;
            while (i < m && j < n) {
                if (nums1[i] == nums2[j]) {
                    ans[k++] = nums1[i];
                    i++;
                    j++;
                } else if (nums1[i] < nums2[j])
                    i++;
                else if (nums2[j] < nums1[i])
                    j++;
            }        
    return Arrays.copyOfRange(ans, 0, k);
        }
    }