Array.prototype.flat()

248 阅读1分钟

flat() 方法会按照一个可指定的深度递归遍历数组,并将所有元素与遍历到的子数组中的元素合并为一个新数组返回。接收指定要提取嵌套数组的结构深度,默认值为 1。返回 一个包含将数组与子数组中所有元素的新数组。

const arr1 = [0, 1, 2, [3, 4]]
console.log(arr1.flat()) // [0, 1, 2, 3, 4]

const arr2 = [0, 1, 2, [[[3, 4]]]]
console.log() // [0, 1, 2, [3, 4]]
var arr1 = [1, 2, [3, 4]];
arr1.flat();
// [1, 2, 3, 4]

var arr2 = [1, 2, [3, 4, [5, 6]]];
arr2.flat();
// [1, 2, 3, 4, [5, 6]]

var arr3 = [1, 2, [3, 4, [5, 6]]];
arr3.flat(2);
// [1, 2, 3, 4, 5, 6]

//使用 Infinity,可展开任意深度的嵌套数组
var arr4 = [1, 2, [3, 4, [5, 6, [7, 8, [9, 10]]]]];
arr4.flat(Infinity);
// [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

// flat() 方法会移除数组中的空项
var arr5 = [1, 2, , 4, 5];
arr5.flat();
// [1, 2, 4, 5]

实现Array.prototype.flat

reduce + concat + isArray + 递归

Array.prototype.deepFlat = function() {
    let arr = this
    let deep = arguments[0] !== Infinity ?
        arguments[0] >>> 0 :
        Infinity
    
    return deep > 0 ?
            arr.reduce(function(prev, cur) {
                return prev.concat(Array.isArray(cur) ? cur.deepFlat(deep - 1) : cur)
            }, []) : arr
}

forEach + isArray + push + 递归

Array.prototype.deepFlat = function() {
    let arr = this
    let deep = arguments[0] !== Infinity ?
        arguments[0] >>> 0 : 
        Infinity
    let res = []
    
    (function _(arr, deep) {
        arr.forEach(function(item){
            if (Array.isArray(item)) {
                _(item, deep - 1)
            } else {
                res.push(item)
            }
        })
    })(arr, deep)
    
    return res
}

栈 pop + push

Array.prototype.deepFlat = function() {
    let arr = this
    let stack = [...arr]
    let res = []
    
    while(stack.length) {
        let item = stack.pop()
        if (Array.isArray(item)) {
            stack.push(...item)
        } else {
            res.push(item)
        }
    }
    return res.reverse()
}