跳至主要內容

4. 数组扁平化

鸭梨小于 1 分钟

4. 数组扁平化

数组扁平化就是将 [1, [2, [3]]] 这种多层的数组拍平成一层 [1, 2, 3]。使用 Array.prototype.flat 可以直接将多层数组拍平成一层:

[1, [2, [3]]].flat(2)  // [1, 2, 3]

ES5 实现:

function flatten(arr) {
  let result = []
  for (let i = 0, len = arr.length; i < len; i++) {
    if (Array.isArray(arr[i]))
      result = result.concat(flatten(arr[i]))
    else
      result.push(arr[i])
  }
  return result
}

ES6 实现:

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

  return arr
}

新增 我的更简单实现:

  return arr.reduce((pre, cur) => pre.concat(Array.isArray(cur) ? my_flatten(cur) : cur), [])
}