面试中经常会遇到一道扁平化数组的题目：写一个函数，将输入数组 [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);