两个数组的交集 II
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2] 输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出: [4,9]
说明:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
解题思路:
用一个Map来把nums1存放,键存放nums1中的数字,值存放nums1中的数出现的次数
遍历nums2,遇到重复的部分就放入一个ArrayList中。
把ArrayList中的数放到数组中返回。
代码如下:
public static int[] intersect(int[] nums1, int[] nums2) {
HashMap <Integer,Integer> hm= new HashMap<>();
ArrayList<Integer> arr= new ArrayList<>();
for(int i=0;i<nums1.length;i++){
if(hm.containsKey(nums1[i])) { //遍历数组,如果有重复的就放进map中,在map中记录出现次数
hm.replace(nums1[i], hm.get(nums1[i])+1);
}
else {
hm.put(nums1[i], 1);
}
}
for(int i=0;i<nums2.length;i++) {
if(hm.containsKey(nums2[i]) && hm.get(nums2[i])>0) { //用map与nums2进行比较,有重复的话就放进arr中
arr.add(nums2[i]);
hm.replace(nums2[i], hm.get(nums2[i])-1);
}
}
int [] res= new int[arr.size()];
for(int k=0;k<res.length;k++){
res[k]=arr.get(k);
}
return res;
}