350.两个数组的交集 II

[chasingsnail]

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

 

示例 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 的元素存储在磁盘上，内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

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

思路

查找表

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersect = function(nums1, nums2) {
    // 获取长度较短的数组
    const base = nums1.length > nums2.length ? nums2 : nums1
    const compare = nums1.length > nums2.length ? nums1 : nums2
    let map = {}
    let result = []
    base.forEach(item => {
        map[item] = map[item] ? ++map[item] : 1
    })

    compare.forEach(item => {
        if (map[item]) {
            map[item]--
            result.push(item)
        }
    })
    return result
};

双指针

可将数组排序后，利用双指针解法，分别指向 nums1 和 nums2 的首位。开始遍历，若前者小于后者，则前者向后移动，否则后者移动，找到相等的值则为二者交集。遍历至一方数组结束后。

const intersect = (nums1, nums2) => {
  nums1.sort((a, b) => a - b);
  nums2.sort((a, b) => a - b); // 先排序，使得重复的元素相邻出现
  const res = [];
  let p1 = 0;                  // 两个指针
  let p2 = 0;
  while (p1 < nums1.length && p2 < nums2.length) { // 用&& 因为有一个越界了就不能找交集
    if (nums1[p1] > nums2[p2]) {        // p1指向的数大，移动p2，期待遇到一样大的
      p2++;
    } else if (nums1[p1] < nums2[p2]) { // 类似
      p1++;
    } else {                   // 遇到相同的，推入res数组，两个指针同时移动考察下一个
      res.push(nums1[p1]);
      p1++;
      p2++;
    }
  }
  return res;
};