前言
数组扁平化是指将多层嵌套的数组转换为一个一维数组的过程。在多层嵌套的数组中,元素可以是基本数据类型,也可以是数组,这种层层嵌套的结构使得数组的处理变得复杂。通过扁平化数组,我们可以简化对数组的操作和处理。
扁平化数组的概念在处理数据时非常有用,特别是在处理递归结构或者在进行某些操作时,我们更希望使用一维数组而不是多维数组。在前端开发中,对数组进行扁平化通常用于处理递归的组件结构,处理嵌套的API响应数据,或者在某些数据操作中简化数据结构。
举个例子,有以下多层嵌套的数组:
const nestedArray = [1, [2, [3, 4], 5]];
这个数组包含了数字和嵌套的数组。扁平化这个数组的结果将是:
const flattenedArray = [1, 2, 3, 4, 5];
下面来介绍一下JS中扁平化数组的几个方式:
递归方法:
使用递归函数来遍历数组,将嵌套的数组展开。这是一种比较直观的方法,但需要小心处理嵌套层次较深的情况,以防止栈溢出。
function flattenArray(arr) {
let flattened = [];
for (let i = 0; i < arr.length; i++) {
if (Array.isArray(arr[i])) {
flattened = flattened.concat(flattenArray(arr[i]));
} else {
flattened.push(arr[i]);
}
}
return flattened;
}
const nestedArray = [1, [2, [3, 4], 5]];
const flattenedArray = flattenArray(nestedArray);
console.log(flattenedArray); // [1, 2, 3, 4, 5]
使用 reduce()
function flattenArray(arr) {
return arr.reduce((acc, val) => {
return acc.concat(Array.isArray(val) ? flattenArray(val) : val);
}, []);
}
const nestedArray = [1, [2, [3, 4], 5]];
const flattenedArray = flattenArray(nestedArray);
console.log(flattenedArray); // [1, 2, 3, 4, 5]
使用flat
flat() 方法是ES2019引入的,它可以用来将嵌套数组拉平指定的层数。
Array.prototype.flat([depth]) 方法是用于将嵌套数组拉平指定层数的数组方法。该方法可以接受一个可选的整数参数,用于指定要拉平的嵌套层数。默认值是 1。如果你传递的是一个大于或等于 0 的整数,则会按照指定的层数进行拉平。如果传递的是Infinity,则会完全拉平所有嵌套的数组。
例如:
const nestedArray = [1, [2, [3, 4], 5]];
// 拉平一层
const flatArray1 = nestedArray.flat(); // [1, 2, [3, 4], 5]
// 拉平两层
const flatArray2 = nestedArray.flat(2); // [1, 2, 3, 4, 5]
// 完全拉平
const fullyFlattenedArray = nestedArray.flat(Infinity); // [1, 2, 3, 4, 5]
数组的flat函数确实很受人喜欢,毕竟这么好用的方法。
使用栈的方式
基本思路是遍历数组,将数组中的元素逐个放入栈中,同时检查每个元素是否为数组。如果是数组,就继续遍历这个数组;如果不是数组,就将元素放入结果数组中。这样,通过迭代和栈的结构,可以实现数组的扁平化。
function flattenArrayWithStack(arr) {
const stack = [...arr];
const result = [];
while (stack.length) {
const next = stack.pop();
if (Array.isArray(next)) {
stack.push(...next);
} else {
result.unshift(next);
}
}
return result;
}
const nestedArray = [1, [2, [3, 4], 5]];
const flattenedArray = flattenArrayWithStack(nestedArray);
console.log(flattenedArray); // [1, 2, 3, 4, 5]