题目描述
给定两个数组,编写一个函数来计算它们的交集。
示例 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 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
思路:先进行排序,然后遍历获取交集即可
代码如下:
public int[] intersect(int[] nums1, int[] nums2) {Arrays.sort(nums1);Arrays.sort(nums2);return intersect1(nums1.length>=nums2.length?nums1:nums2, nums1.length>=nums2.length?nums2:nums1);}private int[] intersect1(int[] longArr, int[] shortArr) {int[] res = new int[shortArr.length];int count=0, j=0;for(int i=0;i<shortArr.length;i++) {for(;j<longArr.length;j++) {if(shortArr[i] == longArr[j]) {res[count++]=longArr[j++];break;}else if (shortArr[i] < longArr[j]) {break;}}}return Arrays.copyOf(res, count);}
改进一下,使用单层循环
public int[] intersect(int[] nums1, int[] nums2) {Arrays.sort(nums1);Arrays.sort(nums2);return intersect1(nums1.length>=nums2.length?nums1:nums2, nums1.length>=nums2.length?nums2:nums1);}private int[] intersect1(int[] longArr, int[] shortArr) {int[] res = new int[shortArr.length];int count=0;for(int i=0, j=0;i<shortArr.length && j<longArr.length;i++) {if (shortArr[i] > longArr[j]) {while (shortArr[i] > longArr[j] && j<longArr.length-1) {j++;}}if(shortArr[i] == longArr[j]) {res[count++]=longArr[j++];}}return Arrays.copyOf(res, count);}