数组扁平化的几种方式及其优缺点

前言

无论是做项目还是找工作,数组扁平化都是一种重要的知识技巧
在2019年发布的ES10中,js更是对数组新增了扁平化的API:Array.protoytype.flat
下面将给出flat实现的几种方式,顺便谈谈这些方式的优缺点

一、转成字符串处理

思路是将数组转成成字符串处理,通过join或者正则替换的方式过滤掉中括号
join是个神奇的api, 它将数组转换成字符串时会剔除掉串中所有的中括号, amazing~

function flatten1(arr) {
    if (!Array.isArray(arr)) return [] 
    return Array.prototype.join.call(arr).split(',')
}
function flatten2(arr1) {
    return JSON.stringify(arr1).replace(/\[|\]/g, '').split(',')
}

此招胜在简单粗暴,用起来大呼过瘾,但同时也败在简单粗暴 ———— 并非所有类型在整容成字符串之后还能完美的还原
拿序列化举例,JSON是一种常用的数据传输格式,对常用的数据载体如对象、数组的支持较好,而对于其他引用类型如Map、Set、Date等的支持欠佳,因此在使用序列化实现flat或者深复制的时候,需要诸位根据实际场景来判断使用

适合数组元素全是字符串或者数字的场景下使用

二、递归实现flat

转成字符串确然用起来爽,但是也存在支持的类型有限的问题
因此,我奉劝各位年轻人,要讲武德,不要犯这样的聪明,啊,小聪明,码林要以勤恳踏实为贵,大家还是老老实实地跟着马保国老师学习正统武术吧~

function flatten3(arr) {
    let res = []
    arr.forEach(v => Array.isArray(v) ? res = res.concat(flatten3(v)) : res.push(v))
    return res
}

以上实现,思路还算清晰,但似乎还可以尝试一下更加简洁的实现方式

function flatten4(arr) {
    return Array.prototype.reduce.call(arr, (total, cur) => total.concat(Array.isArray(cur) ? flatten4(cur) : cur), [])
}

递归方式实现可以避免字符串方式类型不支持的问题,但每当函数调用时都会在栈里储存一些信息,如参数、局部变量等等,递归时函数执行次数过多会抛出栈溢出异常,因此探讨一下其他的实现方式也是很有必要的

三、非递归实现flat

这里尝试用while循环+队列实现一下非递归的flat

function flatten5(arr) {
    let res = []
    while (true) {
        if (arr.length === 0) break
        let cur = arr.shift()
        Array.isArray(cur) ? arr.unshift(...cur) : res.push(cur)
    }
    return res
}

四、ES10 flat 实现指定深度的数组扁平化

ES10中,官方提供的原生flat方法支持传入一个depth参数,这里尝试实现一下

function flatten6(arr,depth = 1) {
    if (!Array.isArray(arr)) return []
    if(typeof depth !== 'number') depth = 1
    let curDepth = 1  
    function flat(arr,curDepth){
        return Array.prototype.reduce.call(arr, (total, cur) =>{
            // 无需扁平化的场景: 超过深度阈值 或 当前元素非数组
            if(curDepth > depth || !Array.isArray(cur)){
                total.push(cur)
                return total
            }else{
                curDepth ++
                return total.concat(flat(cur,curDepth))
            }
        }
        , [])
    }
    return flat(arr,curDepth)
}

测试

猜你喜欢

转载自blog.csdn.net/qq_41109610/article/details/113577706