flat 函数第一个参数为需要扁平化的数组,第二个参数为需要打平的层数,默认不传则打平一层,传入几则代表打平几层。
测试用例
const arr = [1, [2], [3, [4, [5]]]];
console.log(flat(arr));
// [1, 2, 3, [4,[5]]]
console.log(flat(arr, 1));
// [1, 2, 3, [4,[5]]]
console.log(flat(arr, 2));
// [1, 2, 3, 4,[5]]
console.log(flat(arr, 3));
// [1, 2, 3, 4, 5]
js 实现
// 方法一
function flat(arr, level) {
if (!level || level === 1) {
return arr.reduce((a, b) => {
return a.concat(b);
}, []);
} else {
return arr.reduce((a, b) => {
if (Array.isArray(b) && level > 0) {
return a.concat(flat(b, level--));
} else {
return a.concat(b);
}
}, []);
}
}
// 方法二
function flat(arr, depth = 1) {
let result = [];
for (let i = 0; i < arr.length; i++) {
if (Array.isArray(arr[i]) && depth > 0) {
result.push(...flat(arr[i], depth - 1));
} else {
result.push(arr[i]);
}
}
return result;
}