前端面试题 数组 Flat 数组扁平化

数组扁平化： 用于将嵌套多层的数组“拉平”，变成一维的数组 一、[1, [2]] => [1,2]
二、[[1, 2], [3, 4]] => [1, 2, 3, 4]
三、[1, 2, [3, 4, [5, 6]]] => [1, 2, 3, 4, 5, 6]
复制代码

方法一：

使用ES2019的 Array.prototype.flat 函数, 该函数接收一个number类型的参数，表示数组扁平化的深度。

let arr= [2, 3, 4, 5, [21, 22, 24, 25, [31, 33, 34, 35, [41]], 26], 9]
arr.flat(1)
// [2, 3, 4, 5, 21, 22, 24, 25, [31, 33, 34, 35, [41]], 26, 9]
arr.flat(2)
// [2, 3, 4, 5, 21, 22, 24, 25, 31, 33, 34, 35, [41], 26, 9]
arr.flat(4)
// [2, 3, 4, 5, 21, 22, 24, 25, 31, 33, 34, 35, 41, 26, 9]
复制代码

方法二：

使用递归的方式来扁平化数组：

let flat: (arr: Array<any>) => Array<any> 
flat = function (arr: Array<any>): Array<any>{
  let newArr: Array<any> = []
  arr.map((val: any) => {
    if (Array.isArray(val)) {
      newArr = newArr.concat(flat(val))
    } else {
      newArr.push(val)
    }
  })
  return newArr
}

let arr= [2, 3, 4, 5, [21, 22, 24, 25, [31, 33, 34, 35, [41]], 26], 9]

flat(arr)
// [2, 3, 4, 5, 21, 22, 24, 25, 31, 33, 34, 35, 41, 26, 9]
复制代码

该方法有一个缺点，就是数组嵌套层级过多时会抛出栈溢出的错误，那么如何解决这个问题呢？ 请看方法三。

方法二：

使用广度优先（队列）的方式来扁平化数组：

let flat: (arr: Array<any>) => Array<any>
flat = function (arr: Array<any>) {
  let newArr: Array<any> = []
  let queue: Array<any> = []
  queue.push(arr)
  while(queue.length) {
    let item = queue.shift()
    item.map((val) => {
      if (Array.isArray(val)) {
        queue.push(val)
      } else {
        newArr.push(val)
      }
    })
  }
  return newArr
}

let arr= [2, 3, 4, 5, [21, 22, 24, 25, [31, 33, 34, 35, [41]], 26], 9]

flat(arr)
// [2, 3, 4, 5, 21, 22, 24, 25, 31, 33, 34, 35, 41, 26, 9]
复制代码

使用这种方法的话不管数组层级再深，也不会报错。