数组的扁平化,就是将一个嵌套多层的数组 array (嵌套可以是任何层数)转换为只有一层的数组。
var arr = [1, [2, [3, 4]]];
//扁平化之后:[1,2,3,4];
- 递归
循环数组元素,如果还是一个数组,就递归调用该方法
function flatten(arr){
var res = [];
for(var i=0;i<arr.length;i++){
if(Array.isArray(arr[i])){
res = res.concat(flatten(arr[i]));
}else{
res.push(arr[i]);
}
}
return res;
}
- reduce
使用reduce来简化代码:
function flatten(arr){
return arr.reduce(function(prev,item){
return prev.concat(Array.isArray(item)?flatten(item):item);
},[]);
}
- toString
如果数组的元素都是数字,那么我们可以考虑使用 toString 方法,因为:
toString会将数组中的数以逗号形式结合起来。
[1, [2, [3, 4]]].toString() // "1,2,3,4"
toString之后再split转成数组,并将其转换回数字:
function flatten(arr){
return arr.toString().split(',').map(function(item){
return +item;
})
}
但是这种场景只适用于数组内全部是数字的情况,因为中间是全部转换为字符串了。
- ES6扩展运算符
function flatten(arr){
while(arr.some(item=>Array.isArray(item)){
arr = [].concat(...arr);
}
return arr;
}
由于扩展运算符一次只能展开一层数组:
var arr = [1, [2, [3, 4]]];
console.log([].concat(...arr)); // [1, 2, [3, 4]]
因此考虑只要数组中还有数组,就使用扩展运算符展开一次。