面试中经常会遇到一道扁平化数组的题目:写一个函数,将输入数组 [1, [2, [3, [4]], 5]]
转化为 [1, 2, 3, 4, 5]
的形式。
一开始,我以为是通过此题来考察递归的使用,后来仔细一想,事情远远不止这么简单,本文试着通过几种方式的实现,来理解此题可能要考察的内容。
递归
递归是首选思路,一般人都会想到这个方法。
function flat(arr) {
var result = [];
for (var i = 0; i < arr.length; i++) {
if (arr[i] instanceof Array) {
result = result.concat(arguments.callee(arr[i]));
} else {
result.push(arr[i]);
}
}
return result;
}
隐式类型转换
主要考察数组toString()
、valueOf()
方法的使用:
// 借用数组的toString方法
function flat(arr) {
var str = arr.toString();
return str.split(',');
}
// 借用数组的valueOf方法
Array.prototype.valueOf = function () {
return this.join(',');
};
function flat2(arr) {
var str = arr.valueOf();
return str.split(',');
}
Generator
考察ES6中Generator
函数的使用,以及返回值作为一个遍历器对象的使用,以及数组…(扩展运算符)的使用。
function* _flat(arr) {
for (let i = 0; i < arr.length; i++) {
const item = arr[i];
if (Array.isArray(item)) {
yield* _flat(item);
} else {
yield item;
}
}
}
const flat = (arr) => [..._flat(arr)];
Symbol.iterator
考察ES6中Symbol.iterator
的使用
Array.prototype[Symbol.iterator] = function () {
let arr = [].concat(this);
let first = function (arr) {
return arr.shift();
};
return {
next: function () {
let item = first(arr);
if (item) {
return { value: item.toString(), done: false };
} else {
return { done: true };
}
},
};
};
const flat = (arr) => {
let res = [];
for (let i of arr) {
res.push(i);
}
return res.join(',').split(',');
};
或者通过generator函数+递归实现
Array.prototype[Symbol.iterator] = function* (array) {
let arr = array || this;
let sel = arguments.callee;
for (let i = 0; i < arr.length; i++) {
const item = arr[i];
if (Array.isArray(item)) {
yield* sel(item);
} else {
yield item;
}
}
};
// test
var arr = [1, [2, [3, [4]], '5']];
console.log([...arr]);
扩展运算符
function flatten(arr) {
var arr;
while (arr.some((v) => Array.isArray(v))) {
arr = [].concat(...arr);
}
return arr;
}
Array.prototype.flat()
flat()
是ES6中新增的数组方法,该方法返回一个新数组,对原数据没有影响。
[1, [2, [3, [4]], 5]].flat(Infinity);
这篇文章目前没有评论