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

桃扇骨 2024-04-17 06:31 117阅读 0赞

数组扁平化： 用于将嵌套多层的数组“拉平”，变成一维的数组 1、[1, [2]] => [1,2]
    2、[[1, 2], [3, 4]] => [1, 2, 3, 4]
    3、[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]
    复制代码

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

转载于:https://juejin.im/post/5d24abc8518825121c005352