[leetcode]两个数组的交集[javascript]

https://leetcode-cn.com/problems/intersection-of-two-arrays/

描述

给定两个数组，编写一个函数来计算它们的交集。
示例 1:

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

示例 2:

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

说明:

• 输出结果中的每个元素一定是唯一的。
• 我们可以不考虑输出结果的顺序。

分析

先找到两个数组中较小的一个
然后将较小的遍历，与另一个数组进行indexOf，存在就push到新的数组，最后转换set去重

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function(nums1, nums2) {
  let big = nums1, small = nums2, arr = []
  if (nums1.length > nums2.length) {
    big = nums1
    small = nums2
  } else {
    big = nums1
    small = nums2
  }
  for (let i = 0; i < small.length; i++) {
    if (big.indexOf(small[i]) !== -1) {
      arr.push(small[i])
    }
  }
  return Array.from(new Set(arr))
};

评论中找到的使用es6的filter的做法，很酷

var intersection = function(nums1, nums2) {
   return [... new Set(nums1.filter(item => nums2.includes(item)))]
};

相关链接

两个数组的交集 II