写个方法找出两个二维数组的差集

Riesenzahn
127 阅读3分钟

"```markdown 在处理二维数组时，我们常常需要找出两个数组之间的差集。差集的意思是找出属于第一个数组但不属于第二个数组的元素。这里我们将讨论如何在 JavaScript 中实现这一功能。

方法：找出两个二维数组的差集

我们假设我们有两个二维数组，分别为 array1 和 array2，目标是找出 array1 中存在但 array2 中不存在的所有元素。可以通过以下步骤实现：

  1. 将二维数组扁平化
    首先，我们需要将二维数组转为一维数组，这样才能更方便地比较差集。JavaScript 中可以通过 Array.prototype.flat() 方法来扁平化数组。

  2. 使用 Set 来优化查找效率
    为了提高查找效率，我们可以将 array2 转换为一个 Set。Set 对象是一个无重复值的数据集合，其查找元素的时间复杂度为 O(1)，因此可以加速差集查找。

  3. 遍历并找出差集
    我们遍历 array1 中的每一个元素，检查它是否在 array2 中存在，如果不存在则加入差集结果。

代码实现

function findDifference(array1, array2) {
  // 扁平化数组
  const flatArray1 = array1.flat();
  const flatArray2 = array2.flat();
  
  // 将 array2 转换为 Set 来提高查找效率
  const setArray2 = new Set(flatArray2);
  
  // 过滤出 array1 中存在，但 array2 中不存在的元素
  const difference = flatArray1.filter(item => !setArray2.has(item));

  return difference;
}

// 示例
const array1 = [
  [1, 2, 3],
  [4, 5],
  [6]
];

const array2 = [
  [2, 3],
  [6, 7],
  [8]
];

const diff = findDifference(array1, array2);
console.log(diff); // 输出 [1, 4, 5]

代码解析

  1. 扁平化数组：
    flat() 方法将多维数组转换为一维数组。array1.flat() 和 array2.flat() 将 array1 和 array2 扁平化，得到一维数组 flatArray1 和 flatArray2。

  2. 使用 Set：
    将 array2 扁平化后的数组 flatArray2 转换为 Set，命名为 setArray2。这样，后续对元素的查找可以利用 Set 的 O(1) 时间复杂度。

  3. 过滤差集：
    使用 filter() 方法遍历 flatArray1，对于每个元素 item，通过 setArray2.has(item) 检查该元素是否存在于 setArray2 中。如果不存在，则将其加入到差集数组中。

性能分析

  • 扁平化数组的时间复杂度是 O(n)，其中 n 为数组中所有元素的数量。
  • 将 array2 转换为 Set 的时间复杂度是 O(m)，其中 m 为 array2 扁平化后元素的数量。
  • 使用 filter() 遍历 array1 中每个元素的时间复杂度是 O(n)，而 Set 中的查找操作是 O(1)。

因此，总体的时间复杂度是 O(n + m)，空间复杂度是 O(m)，这是一个较为高效的算法。

扩展：处理对象类型的差集

如果数组中包含的是对象而非基本类型的数据，处理起来稍微复杂一些。因为对象在 JavaScript 中是引用类型，直接使用 Set 的 has() 方法无法正确比较对象是否相同。我们可以借助 JSON 字符串化来处理对象类型的差集。

示例代码

function findObjectDifference(array1, array2) {
  // 扁平化数组
  const flatArray1 = array1.flat();
  const flatArray2 = array2.flat();
  
  // 将 array2 转换为 Set，并通过 JSON 字符串化来处理对象
  const setArray2 = new Set(flatArray2.map(item => JSON.stringify(item)));
  
  // 过滤出 array1 中存在，但 array2 中不存在的对象
  const difference = flatArray1.filter(item => !setArray2.has(JSON.stringify(item)));

  return difference;
}

// 示例
const array1 = [
  [{ id: 1, name: 'A' }, { id: 2, name: 'B' }],
  [{ id: 3, name: 'C' }]
];

const array2 = [
  [{ id: 2, name: 'B' }],
  [{ id: 3, name: 'C' }]
];

const diff = findObjectDifference(array1, array2);
console.log(diff); // 输出 [{ id: 1, name: 'A' }]

总结

通过以上实现，我们能够有效地找出两个二维数组的差集。我们首先扁平化数组，然后使用 Set 来提高查找效率，最后通过 filter() 找出 array1 中存在但 array2 中不存在的元素。对于包含对象的数组，我们可以通过 JSON 字符串化的方式处理对象差集。

这种方法适用于数组元素为基本类型或对象类型的差集查找，并且性能较为优越，特别是在处理大规模数据时非常高效。

avatar
文章
阅读
粉丝