给定两个数组,写一个方法来计算它们的交集。
例如:
给定 nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].
注意:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
跟进:
- 如果给定的数组已经排好序呢?你将如何优化你的算法?
- 如果 nums1 的大小比 nums2 小很多,哪种方法更优?
- 如果nums2的元素存储在磁盘上,内存是有限的,你不能一次加载所有的元素到内存中,你该怎么办?
分析:
349. 两个数组的交集的进阶版,
349 是要求无重复元素,该题可以有重复元素,考虑换种方法
先将两个数组排序,再对俩数组分别设置遍历指针,进行循环判断,如:
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);
}
}