(leetcode.350)
题目要求
给你两个整数数组 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]
思路
使用HashMap,key为交集数字,value为每个数字出现次数。
配合int[] result,当HashMap中value变化时,记录当前值作为交集元素。
注意:
1.最终返回的数组长度一定 <= 两个数组的最小长度。
2.先遍历短数组,每个元素个数自增。再遍历长数组,符合条件(>0)的元素个数递减。
代码
public int[] intersect(int[] nums1, int[] nums2) {
if(nums1.length > nums2.length) return intersect(nums2,nums1);//第一个短,第二个长
Map<Integer,Integer> map = new HashMap<Integer,Integer>();
int[] result = new int[nums1.length];
for(int i : nums1) {//短数组赋值
int n = map.getOrDefault(i,0) + 1;
map.put(i,n);
}
int index = 0;
for(int i = 0; i < nums2.length; i ++) {
int temp = nums2[i];
if(map.getOrDefault(temp,0) != 0) {
int n = map.get(temp) - 1;
map.put(temp,n);
result[index ++] = temp;
}
}
return Arrays.copyOfRange(result,0,index);
}
copyOfRange()方法
copyOfRange(original,int a,int b):返回一个从a位置到b位置的数组
举例:
int[] n=new int[]{1,2,3,4,5};
int[] m=Arrays.copyOfRange(n, 0, 3);