给你两个整数数组 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]
提示:
1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000
解法:
定义哈希表,key存储数字,value存储出现次数
遍历nums1,存入哈希表
遍历nums2,对比哈希表中的value,若value为正数,则value–,将该key加入结果数组中
边界情况:
结果数组的结尾可能后面有0(定义结果数组时使用了nums1的长度):使用Arrays.copyOfRange(ans,0,index),返回实际使用的长度
代码:
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
if (nums1.length>nums2.length)
{
intersect(nums2,nums1);
}
int [] ans=new int[nums1.length];
HashMap<Integer, Integer> map = new HashMap<>();
for (int j : nums1) {
if (map.containsKey(j)) {
int temp = map.get(j);
temp++;
map.put(j, temp);
} else {
map.put(j, 1);
}
}
int index=0;
for (int i:nums2)
{
if (map.containsKey(i))
{
int temp=map.get(i);
if (temp>0)
{
temp--;
map.put(i,temp);
ans[index]=i;
index++;
}
}
}
return Arrays.copyOfRange(ans,0,index);
}
}