给定两个数组,编写一个函数来计算它们的交集。
示例 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.com/problems/intersection-of-two-arrays-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* @param {number[]} nums1
* @param {number[]} nums2
* @return {number[]}
*/
var intersect = function(nums1, nums2) {
let map1=makeCountMap(nums1);
let map2=makeCountMap(nums2);
let res=[];
for(let num of map1.keys()){
let count1=map1.get(num);
let count2=map2.get(num);
if(count2){
let pushCount=Math.min(count1,count2);
for(let i=0;i<pushCount;i++){
res.push(num);
}
}
}
return res;
};
function makeCountMap(nums){
let map=new Map();
for(let i=0;i<nums.length;i++){
let num=nums[i];
let count=map.get(num);
if(count){
map.set(num,count+1);
}else{
map.set(num,1);
}
}
return map;
}
let nums1 = [4,9,5];
let nums2 = [9,4,9,8,4];
console.log(intersect(nums1,nums2))
console.log(intersect([1,2,2,1],[2,2]))
makeCountMap()的作用为根据数组,转化为以下map集合,key为数组中元素,value是元素在数组中出现的次数。
首先通过makeCountMap()函数,将num1和num2两个数组转换成两个以上类型的map集合,遍历map1的keys()方法,通过get方法得到map1与map2在各自数组中出现的次数count1与count2,由于for循环是遍历的map1,所以只需要判断map2中的key值count2是否存在,如果存在则表示此key值在两个数组中有交集, 通过比对map1与map2中的出现的次数,选择出次数最少的pushCount,将此map1循环中的key值添加到res数组当中。所得res数组则为两数组的交集