使用flat()方法实现数组扁平化和手写数组扁平化函数

使用flat()方法实现数组扁平化和手写数组扁平化函数

flat()方法

flat()方法会按照一个可指定的深度递归遍历数组，并将所有元素与遍历到的子数组中的元素合并为一个新数组返回，对原数据没有影响。
下面为flat方法使用的案例：

const testArr=["a",["b","c"],["d",["e",["f"]]],"g"];
//flat不传参时，默认扁平化一层
console.log(testArr.flat());
// [ 'a', 'b', 'c', 'd', [ 'e', [ 'f' ] ], 'g' ]

//flat传入一个整数参数，整数即扁平化的层数
console.log(testArr.flat(2));
// [ 'a', 'b', 'c', 'd', 'e', [ 'f' ], 'g' ]

// Infinity 关键字作为参数时，无论多少层嵌套，都会转为一维数组
console.log(testArr.flat(Infinity));
// ["a", "b", "c", "d", "e", "f", "g"]

// 传入 <=0的整数将返回原数组，不扁平化
console.log(testArr.flat(0));
console.log(testArr.flat(-1));
// ["a", ["b", "c"], ["d", ["e", ["f"]], "g"]]

// 如果原数组有空位，flat()方法会跳过空位
console.log(["a","b","c",,"d"].flat());
// [ 'a', 'b', 'c', 'd' ]

用函数实现数组扁平化

方法一、使用reduce方法

// 一次扁平化所有
const flattenDeep = (arr) =>{
    return Array.isArray(arr) 
    ? arr.reduce((acc,cur) => [...acc,...flattenDeep(cur)],[]) 
    : [arr]
}
let testArr = ["a",["b","c"],["d",["e",["f"]]],"g"];
console.log(flattenDeep(test));
// ["a", "b", "c", "d", "e", "f", "g"]

// 实现flat函数：
const flat = (arr,depth=1) =>{
    return depth > 0
    ? arr.reduce((acc,cur)=>{
        if(Array.isArray(cur)){
            return [...acc,...flat(cur,depth - 1)]
        }
        return [...acc,cur]
    },[])
    : arr
}
console.log(flat(testArr,2));
// [ 'a', 'b', 'c', 'd', 'e', [ 'f' ], 'g' ]

方法二、栈

// 一次性降维所有
const flattenDeep = (arr) =>{
    const result = [];
    //将数组元素拷贝至栈，直接赋值会改变原数组 
    const stack = [...arr];
    // 如果栈不为空，则循环遍历
    while(stack.length !== 0){
        // 删除最后一个元素，并返回它
        const val = stack.pop();
        if(Array.isArray(val)){
            // 如果是数组，则再次入栈，并且展开一层
            stack.push(...val);
        }else{
            // 如果不是数组，就用头插法插入到结果数组中
            result.unshift(val);
        }
    }
    return result;
}

console.log(flattenDeep(testArr));
// ["a", "b", "c", "d", "e", "f", "g"]

参考自：手写数组去重、扁平化函数