前端面试中数组扁平化也是一个比较常见的问题了,我见得多的有两种问法:
一种是将数组完全展开,数组里面不在有数组,另一种就是给定一个展开值,只展开一部分,下面就这两种情况进行一下讨论。
1. 完全展开
先创建一个空数组作为最后的输出结果,然后将要扁平化的数组去进行遍历,如果发现arr[i]是数组,就进行递归,判断这个数组里面的值是否还是数组,直到递归完全,每一次递归完都会将最后的值添加到空数组里面去。
let res = [];
function flat(arr) {
for(let i = 0; i < arr.length; i++) {
if(Array.isArray(arr[i])) {
flat(arr[i])
} else {
res.push(arr[i])
}
}
return res;
}
console.log(flat([1, [2, [3, [4]]]]));
2. 部分展开,给出展开值num
和上面原理相似,只不过每次递归的时候要去判断num是否已经为0,另外因为concat不会改变原数组而是另外创建一个新数组,所以res会需要重新赋值。
function flat(arr, num) {
if (num === 0) {
return arr;
}
const len = arr.length;
let res = [];
for (let i = 0; i < len; i++) {
if (Array.isArray(arr[i])) {
res = res.concat(flat(arr[i], num - 1));
} else {
res.push(arr[i]);
}
}
return res;
}
//测试
console.log(flat([1, [2, [3, [4]]]], 2));
当然,数组扁平化的方法还有很多种,如果toString,展开运算符,reduce等等,感兴趣的小伙伴们可以自行去了解一下。