350. 两个数组的交集 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  的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

解法

1、先对两个数组进行排序

先对两个数组进行排序,然后使用两个指针,分别指向两个数组开始的位置。

  • 如果两个指针指向的值相同,说明这个值是他们的交集,就把这个值加入到集合 list 中,然后两个指针在分别往后移一步。
  • 如果两个指针指向的值不同,那么指向的值相对小的往后移一步,相对大的先不动,然后再比较

一直重复上面的操作,直到其中一个指针不能再移动为止,最后再把集合 list 转化为数组即可

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersect = function (nums1, nums2) {
  nums1.sort((a, b) => a - b);
  nums2.sort((a, b) => a - b);

  let i = 0;
  let j = 0;
  let result = [];

  while (i < nums1.length && j < nums2.length) {
    if (nums1[i] === nums2[j]) {
      result.push(nums1[i]);
      i++;
      j++;
    } else if (nums1[i] < nums2[j]) {
      i++;
    } else {
      j++;
    }
  }
  return result;
};

2、使用map解决

还可以使用map来解决,具体操作如下

  • 遍历nums1中的所有元素,把它存放到map中,其中key就是nums1中的元素,value就是这个元素在数组nums1中出现的次数。

  • 遍历nums2中的所有元素,查看map中是否包含nums2的元素,如果包含,就把当前值加入到集合list中,然后对应的value要减1。

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersect = function (nums1, nums2) {
  const map = {};

  for (let i = 0; i < nums1.length; i++) {
    if (map[nums1[i]] === undefined) {
      map[nums1[i]] = 1;
    } else {
      map[nums1[i]] += 1;
    }
  }

  let result = [];
  for (let j = 0; j < nums2.length; j++) {
    if (map[nums2[j]]) {
      result.push(nums2[j]);
      map[nums2[j]] -= 1;
    }
  }
  return result;
};