针对这样一个数组[3,[4,8,[9,1],3],[6,8],[2,10],5,7]想要将其展开,一般都会想到的是利用递归。
如下这样一个方法:
function flatten(arr) {
var newArr = [];
arr.map(item => {
if(Array.isArray(item)){
newArr.push(...flatten(item))
} else {
newArr.push(item)
}
})
return newArr
}
但是这样的方法在数组嵌套非常深的情况下,函数递归调用时,调用层次太深,就会造成栈溢出。何为栈溢出?
栈是一种后进先出的数据结构,堆栈也是采用这种结构管理内存,调用过程中当最初的结果依赖于后面的计算处理,那么后面的部分虽然后开始处理,却先结束。当后续处理太多并且又依赖更后面的处理......(比如递归),便会一直压栈,当空间全部用完,就会造成“堆栈溢出”。
更通俗一点就是向一个水桶里装水,等装满了之后再加水就溢出了。
这时候就应该换一种方法来解决这个问题,调用数组的toString方法或者join方法,将数组变为字符串然后再用split分割还原为数组可以实现。
function flatten(arr) {
return arr.toString().split(',').map(function(item) {
return Number(item);
})
}
// 或者
function flatten(arr) {
return arr.join(',').split(',').map(function(item) {
return Number(item);
})
}
但是该方法有一个缺点,就是会转换数组中元素item的类型,需要进行类型转换。
推荐使用es6的扩展运算符来实现数组扁平化。
function flatten(arr) {
let stack = [...arr].reverse()
let newArr = []
while(stack.length){
let o = stack.pop()
if(Array.isArray(o)){
stack.push(...o.reverse())
} else {
newArr.push(o)
}
}
return newArr
}