两个数组的交集

两个数组的交集

给定两个数组,编写一个函数来计算它们的交集。

示例 1:

输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]

示例 2:

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
答案

set集合解题法: 使用set集合解题, 判断集合的每一项是否存在另一个集合当中

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function (nums1, nums2) {
 function set_intersection(set1, set2) {
  if (set1.size > set2.size) {
   // 目的 保证长的集合是第一个参数
   return set_intersection(set2, set1);
  }
  const intersection = new Set();
  // 遍历判断是否存在 另一个集合当中
  for (const num of set1) {
   if (set2.has(num)) {
    intersection.add(num);
   }
  }
  return [...intersection];
 }

 let set1 = new Set(nums1);
 let set2 = new Set(nums2);
 return set_intersection(set1, set2);
};

let nums1 = [4, 9, 5];
let nums2 = [9, 4, 9, 8, 4];

const intersection_res = intersection(nums1, nums2);

哈希 解题法:

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function (nums1, nums2) {
 let map = new Map();

 nums1.forEach(item => {
  map.set(item, true);
 });

 return nums2.filter(item => {
  if (map.has(item)) {
   map.delete(item);
   return true;
  }

  return false;
 });
};

const intersection_res = intersection([4, 9, 5], [9, 4, 9, 8, 4, 5]);



本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!