使用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"]
参考自:手写数组去重、扁平化函数