js数组扁平化处理

什么是数组扁平化

数组扁平化就是将一个多维数组转换为一个一维数组。

数组扁平化的实现方式

1.对数组的每一项进行遍历
2.判断该项是否是数组
3.如果不是数组就直接放进新数组
4.是数组就回到1，继续迭代
5.当数组遍历完成，就返回这个新数组

一.常规方式，递归处理

Array.prototype.flatten = function () {
  var resultArr = [];
  var len = this.length;
  for (var i = 0; i < len; i ++) {
    if (Array.isArray(this[i])) {
        resultArr = resultArr.concat(this[i].flatten());
    } else {
      resultArr.push(this[i]);
    }
  }
  return resultArr;
}
var arr=[1,2,3,[4,5,'hello',['world',9,666]]]
console.log(arr.flatten())//[1, 2, 3, 4, 5, "hello", "world", 9, 666]

2.使用reduce方法

Array.prototype.flatten = function () {
  return this.reduce(function (prev, cur, curIndex, arr) {
      return Array.isArray(cur) ? prev.concat(cur.flatten()) : prev.concat(cur); 
  }, []);
}

3.yield的用法，天才级别的扁平化处理

function* iterTree(tree){
      if(Array.isArray(tree)){
          for(let i = 0 ;i<tree.length;i++){
              yield* iterTree(tree[i]);
          }
      }else{
          yield tree;
      }
  }
　let arr1 = [1, 2, ['a', 'b', ['中', '文', [1, 2, 3, [11, 21, 31]]]], 3];
  let list=[]
for (const x of iterTree(arr1)) {
    list.push(x)
  }
  console.log(list);//[1, 2, "a", "b", "中", "文", 1, 2, 3, 11, 21, 31, 3]

4.ES6中的flat也可以实现扁平化处理

let arr1 = [1,2,['a','b',['中','文',[1,2,3,[11,21,31]]]],3];
 console.log( arr1.flat( Infinity ) ); 

5.扩展运算符

//ES6 增加了扩展运算符，用于取出参数对象的所有可遍历属性，拷贝到当前对象之中：

var arr = [1, [2, [3, 4]]];
console.log([].concat(...arr)); // [1, 2, [3, 4]]
//我们用这种方法只可以扁平一层，但是顺着这个方法一直思考，我们可以写出这样的方法：

// 改进
var arr = [1, [2, [3, 4]]];

function flatten(arr) {

    while (arr.some(item => Array.isArray(item))) {
        arr = [].concat(...arr);
    }

    return arr;
}

console.log(flatten(arr))