【LeetCode】349. 两个数组的交集

前言

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/intersection-of-two-arrays/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方式一：哈希表

/**
 * 哈希表
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function(nums1, nums2) {
  const map = new Map()
  const res = []
  for (let i = nums1.length; i--;) {
    map.set(nums1[i], true)
  }
  for (let i = nums2.length; i--;) {
    const temp = nums2[i]
    if (map.get(temp)) {
      map.delete(temp)
      res.push(temp)
    }
  }
  return res
};

方式二：二分查找

/**
 * 二分查找
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function(nums1, nums2) {
  const map = new Map()
  nums2.sort((a, b) => a - b)
  for (let i = nums1.length; i--;) {
    const temp = nums1[i]
    if (search(nums2, temp) && !map.get(temp)) {
      map.set(temp, true)
    }
  }
  return [...map.keys()]
};

var search = function (nums, target) {
  let left = 0, right = nums.length - 1
  while (left < right) {
    const mid = Math.floor(left + (right - left) / 2)
    if (nums[mid] >= target) {
      right = mid
    } else {
      left = mid + 1
    }
  }
  return nums[right] === target
}

方式三：排序 + 双指针

/**
 * 排序 + 双指针
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function(nums1, nums2) {
  nums1.sort((a, b) => a - b)
  nums2.sort((a, b) => a - b)
  let i = 0, j = 0, set = new Set()
  while (i < nums1.length && j < nums2.length) {
    if (nums1[i] === nums2[j]) {
      set.add(nums1[i])
      i++
      j++
    } else if (nums1[i] < nums2[j]) {
      i++
    } else if (nums1[i] > nums2[j]) {
      j++
    }
  }
  return [...set.keys()]
};

总结

多种方式可以解决问题，选择最合适的即可。