两个数组的交集 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]

说明:

  • 输出结果中每个元素出现的次数,应与元素在两个数组中出现次数的最小值一致。
  • 我们可以不考虑输出结果的顺序。

进阶:

  • 如果给定的数组已经排好序呢?你将如何优化你的算法?
  • 如果nums1的大小比nums2小很多,哪种方法更优?
  • 如果nums2的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

题解

解题思路

  • 生成一个map,遍历其中一个数组,以其中的值为key,以该值出现的次数为value存入到map中。
  • 此时遍历另一个数组,将每一个值与map中存储的key进行比较。
  • 如果在map中存在,则说明是交集数,则存入数组,并将map中对应key的value-1
/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
export default function intersect (nums1: number[], nums2: number[]): number[] {
  // 生成map,存储每个值对应的个数
  const map = new Map()
  for(let i = 0; i < nums1.length; i++) {
    let res = map.get(nums1[i])
    map.set(nums1[i], res ? ++res : 1)
  }
  // 遍历长度较小的数组,如果在map中存在,则说明是交集数,则存入数组,并将map中对应值的数量-1
  const result: number[] = []
  for(let i = 0; i < nums2.length; i++) {
    let targetNum = map.get(nums2[i])
    if (targetNum) {
      result.push(nums2[i])
      map.set(nums2[i], --targetNum)
    }
  }
  return result
}
Last Updated:
Contributors: luhaifeng