Example 32
两个数组的交集 II
题目概述:给你两个整数数组nums1 和 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]
解题思路:Example32同Example31,不同点是该题不再局限于输出结果中的每个元素一定是唯一的,因此无需考虑交集元素唯一性的问题,仅需考察是否是交集元素即可。
解题步骤:
1. 将数组nums1、nums2排序。
2. 定义变量length1、length2分别存储两个数组的长度。
3. 定义数组intersection存储两个数组的交集元素,其长度为length1的length2的最小值(两个数组的交集最多为两个数组长度的最小值)。定义变量index、index1、index2分别记录交集数组、nums1、nums2当前考察的元素索引,初值均为0。
4. 定义while循环,若index1 < length1并且index2 < length2,说明当前还未考察完毕,继续进入循环考察,否则说明至少有一个数组已经考察完毕,则退出该循环。并将交集数组中的第0到index之间的元素(交集元素)复制到新数组中并返回。
5. 在while循环中,定义变量num1、num2分别记录两个数组中当前考察的元素。若num1等于num2,则将该元素加入到交集数组中,并将交集数组的考察索引自加一,而后分别将index1、index2自加一,考察下一元素。
6. 若num1小于num2,则表明nums1[index1]元素一定不是交集元素(因为nums2是递增的),则将index1自加一,继续判断下一个元素是否是交集元素,否则说明num1大于num2,表明nums2[index2]元素一定不是交集元素(因为nums1是递增的),则将index2自加一,继续判断下一个元素是否是交集元素。
示例代码如下:
public class ArrayIntersection_ii { /** * 给你两个整数数组 nums1 和 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] * 来源:力扣(LeetCode) * 链接:https://leetcode.cn/problems/intersection-of-two-arrays-ii */ public static void main(String[] args) { ArrayIntersection_ii ai = new ArrayIntersection_ii(); System.out.println(Arrays.toString(ai.intersection(new int[]{1, 2, 2, 1}, new int[]{2, 2}))); // [2, 2] } /** * 个人修改 * * @param nums1 * @param nums2 * @return */ public int[] intersection(int[] nums1, int[] nums2) { Arrays.sort(nums1); Arrays.sort(nums2); int length1 = nums1.length, length2 = nums2.length; int[] intersection = new int[Math.min(length1, length2)]; int index = 0, index1 = 0, index2 = 0; while (index1 < length1 && index2 < length2) { int num1 = nums1[index1], num2 = nums2[index2]; if (num1 == num2) { intersection[index++] = num1; index1++; index2++; } else if (num1 < num2) { index1++; } else { index2++; } } return Arrays.copyOfRange(intersection, 0, index); } }