js实现 flat 扁平函数

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;
}