手写实现数组的扁平化处理

2024-02-15 16:05:58

数组扁平化：理解多维数组如何变身一维数组

引言

在编程世界中，数组是存储数据的常见结构。有时，我们需要将一个多维数组（嵌套数组）转换为一维数组（平面数组）。这个过程称为数组扁平化。本文将深入探讨数组扁平化的各种方法，帮助您掌握这项基本技巧。

方法一：使用 flat() 方法

ES6 引入了 flat() 方法，它专用于数组扁平化。该方法将多维数组展开为一维数组。您可以指定一个可选参数 depth 来控制展平的深度。如果省略 depth，数组将被完全展平。

const arr = [[1, 2], [3, 4], [5, 6]];
console.log(arr.flat()); // 输出： [1, 2, 3, 4, 5, 6]

const arr2 = [[1, 2], [3, [4, 5]], [6, 7]];
console.log(arr2.flat(2)); // 输出： [1, 2, 3, 4, 5, 6, 7]

方法二：使用 reduce() 方法

reduce() 方法也可以用于数组扁平化。它采用一个回调函数和一个初始值。回调函数将数组中的每个元素与初始值组合起来，返回一个新值。这个新值作为下一次迭代的初始值。

const arr = [[1, 2], [3, 4], [5, 6]];
const flattenedArr = arr.reduce((acc, curr) => acc.concat(curr), []);
console.log(flattenedArr); // 输出： [1, 2, 3, 4, 5, 6]

方法三：使用 for 循环

for 循环是实现数组扁平化的一种简单方法。虽然它效率较低，但对于小型数组来说，这是一个不错的选择。

const arr = [[1, 2], [3, 4], [5, 6]];
const flattenedArr = [];
for (let i = 0; i < arr.length; i++) {
  for (let j = 0; j < arr[i].length; j++) {
    flattenedArr.push(arr[i][j]);
  }
}
console.log(flattenedArr); // 输出： [1, 2, 3, 4, 5, 6]

方法四：使用递归

递归是一种高效的数组扁平化方法，但可能导致堆栈溢出。

const arr = [[1, 2], [3, 4], [5, 6]];

function flattenArray(arr) {
  let flattenedArr = [];
  for (let i = 0; i < arr.length; i++) {
    if (Array.isArray(arr[i])) {
      flattenedArr = flattenedArr.concat(flattenArray(arr[i]));
    } else {
      flattenedArr.push(arr[i]);
    }
  }
  return flattenedArr;
}

console.log(flattenArray(arr)); // 输出： [1, 2, 3, 4, 5, 6]