JavaScript专题(八)数组扁平化

hushuo
发布于 2021-1-14 12:09
浏览
0收藏

一、递归
for循环是我们在进行数组操作时最容易想到的,在不考虑时间和空间复杂度的时候,递归应该是一个完美的选择!

示例:

输入 
const arr = [1, [2, [3, 4, { a: 1 }], null], undefined];`
输出 
[ 1, 2, 3, 4, { a: 1 }, null, undefined ]


代码:

function flatten(arr) {
    let res = [];
    for (let 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是做什么的:reduce() 方法对数组中的每个元素执行一个由您提供的reducer函数(升序执行),将其结果汇总为单个返回值。

示例:

输入 
const arr = [1, [2, [3, 4, { a: 1 }], null], undefined];`
输出 
[ 1, 2, 3, 4, { a: 1 }, null, undefined ]


代码:

function flatten(arr) {
    return arr.reduce((prev, next) => {
        // prev表示上一次操作的结果
        return prev.concat(Array.isArray(next) ? flatten(next) : next)
    }, [])
    // 注意reduce的初始值应该是[],否则无法进行拼接
}


注意:

仔细对照方法一和方法二,两者思路完全一致——找到数据类型是数组的子元素,对其进行抹平处理,只不过实现的细节略有不同。

唯一需要注意的地方就是拼接时数据的基本类型要以数组开始。

三、apply+some
既然方法二是方法一的变式,那个方法三也可以说是方法二的变式,但不在采用递归的方式,而是一层一层“拆除”嵌套的方式

先来看看用到的API:

apply:调用一个具有给定this值的函数,以及以一个数组(或类数组对象)的形式提供的参数。
some:测试数组中是不是至少有1个元素通过了被提供的函数测试。它返回的是一个Boolean类型的值
代码:

function flatten(arr) {
    while (arr.some(item => Array.isArray(item))) {
        // 只要存在数组类型的元素,就抹平一层
        arr = [].concat.apply([], arr)
    }
    return arr;
}


注意:

大家疑惑的点主要在apply,其实主要目的就是减少括号

let res = [];
res  = res.concat({});
// 等价于
[].concat.apply([], [{}])


在本例中:

arr = [].concat.apply([], arr);
// 等价于
[].concat(1, [2, [3, 4, { a: 1 }], null], undefined)


四、ES6展开运算符
大家可能业也注意到了,方法一到方法三 我们不断的借助现有方法,以精简我们的代码量,本方法也是如此~

我们利用ES6的展开运算符(用于取出参数对象的所有可遍历属性,拷贝到当前对象之中),继续精简方法三:

代码:

function flatten(arr) {
    while (arr.some(item => Array.isArray(item))) {
        // 只要存在数组类型的元素,就抹平一层
        arr = [].concat(...arr)
    }
    return arr;
}


五、toString
如果前面四个方法是我们可以实现的主流的方式,那么接下来的几个就是我们可以实现的“非主流”方式,直接上代码!

function flatten(arr) {
    return arr.toString().split(',').map(function(item){
        return +item
    })
}


注意:

将数据转换类型存在类型的限制,如果原始数据是这样的:[1, ‘2’],那么就会出现问题

六、正则
假如我们默认了类型会被转换这一缺陷,那么还可以通过更暴力里的方式来将数组扁平化:

function flatten(arr) {
    return JSON.stringify(arr).replace(/\[|\]/g, '').split(',');
}


注意:

如果是纯数组,貌似没什么问题

function flatten(arr) {
    return JSON.parse('[' + JSON.stringify(arr).replace(/\[|\]/g, '').split(',') + ']')
}


但如果数组内存在对象,就需要消除JSON后的后果,这样更严谨一些~

到这里,六种方法算是包括了大部分数组扁平化的实现,那么如果希望将我们的方法“升级”成工具怎么办?这时候我们就要“抄袭”一下loadsh了~

 

七、实现自己的扁平化工具方法
这里我们简化了loadsh里的flatten函数,具体改动我们来看代码:

 /*
 * @private
 * @param {Array} array 需要扁平化的数组
 * @param {number} depth 最多处理几层
 * @param {boolean} [isStrict] 是否严格处理函数
 * @param {Array} [result=[]] 输出的数组
 * @returns {Array}
 */
function flatten(array, depth, isStrict, result) {
    result || (result = [])
    // 边界
    if (array == null) return result;

    for (const value of array) {
        if (depth > 0 && Array.isArray(value)) {
            if (depth > 1) {
                flatten(value, depth - 1, isStrict, result)
            } else {
                result.push(...value); // 只拆1层
            }
        } else if (!isStrict) {
            result[result.length] = value
        }
    }
    return result;
}


该方法提供了扁平化层数

const res = flatten([1, 2, [3, 4, [5, 6]], { a: 1 }, null, undefined], 1, false);
// [ 1, 2, 3, 4, [ 5, 6 ], { a: 1 }, null, undefined ]


该方法提供了扁平化后相反的效果

const res = flatten([1, 2, [3, 4, [5, 6]], { a: 1 }, null, undefined], 1, true);
// [ 3, 4, [ 5, 6 ] ]
const res = flatten([1, 2, [3, 4, [5, 6]], { a: 1 }, null, undefined], 2, true);
// [ 5, 6 ]


isStrict参数打开后,扁平后保留了被暴漏出来的元素,剔除了浅层元素。

写到这里我们即了解了扁平化的处理思路,也有了一定的实现能力,如果你能完全理解上面的代码,相信扁平化这一部分应该难不到你了,我们下一篇文章继续研究loadsh的另一个方法~

收藏
回复
举报
回复
    相关推荐