LeetCode 350：破解两数组交集的巧妙秘诀，彻底掌握集合运算

闲谈

2024-02-14 14:31:31

LeetCode 350：寻找两个数组的交集

背景

在算法的世界中，LeetCode 是一个广受欢迎的在线竞赛平台，汇聚了众多编程爱好者和算法高手。LeetCode 350 题正是其中一道颇具挑战性的题目，它考察了程序员对于数组和集合运算的理解和应用能力。

题目概述

给定两个整数数组 nums1 和 nums2，请找出两个数组的交集。数组中的每个元素可以重复出现。

算法思路

这道题目的核心是找到两个数组中重复出现的元素，并将其作为交集。对于这个问题，我们可以采用两种不同的算法来解决：

哈希表法

哈希表是一种数据结构，它可以快速地查找和插入元素。我们可以使用哈希表来存储数组 nums1 中的元素，然后遍历数组 nums2，如果当前元素在哈希表中存在，则将其加入交集中。

代码示例

const intersect = (nums1, nums2) => {
  const hashTable = {};
  for (const num of nums1) {
    if (hashTable[num]) {
      hashTable[num]++;
    } else {
      hashTable[num] = 1;
    }
  }

  const intersection = [];
  for (const num of nums2) {
    if (hashTable[num] > 0) {
      intersection.push(num);
      hashTable[num]--;
    }
  }

  return intersection;
};

排序后双指针法

首先，我们将两个数组都进行排序。然后，使用两个指针分别指向两个数组的第一个元素。如果两个指针指向的元素相等，则将其加入交集中，并同时将两个指针向后移动一位。如果两个指针指向的元素不相等，则将指向较小元素的指针向后移动一位。重复这个过程，直到其中一个指针到达数组末尾。

代码示例

const intersect = (nums1, nums2) => {
  nums1.sort((a, b) => a - b);
  nums2.sort((a, b) => a - b);

  const intersection = [];
  let i = 0;
  let j = 0;

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

  return intersection;
};