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

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

示例 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);
    }
}

 

你可能感兴趣的:(Java,Test)