数组扁平化

手写flat（数组扁平化）

Array.prototype._flat=function (depth=1){
  function flatten(arr,depth){
    if(depth<=0) return [...arr]
    return arr.reduce((pre,cur)=>{
      if (Array.isArray(cur)) pre.push(...flatten(cur,depth-1)) //是数组，递归
      else pre.push(cur) //不是数组，直接加入pre
      return pre
    },[])
  }
  return flatten(this,depth)
}

const arr=[1,[2,3],[4,[5,6]],7]
console.log(arr._flat()) //[1, 2, 3, 4, [5, 6], 7]
console.log(arr._flat(1)) //[1, 2, 3, 4, [5, 6], 7]
console.log(arr._flat(2)) //[1, 2, 3, 4, 5, 6, 7]
console.log(arr._flat(Infinity)) //[1, 2, 3, 4, 5, 6, 7]

代码简化：

Array.prototype._flat = function (depth = 1) {
  function flatten(arr, depth) {
    return depth <= 0 ?
        [...arr] : arr.reduce((pre, cur) => Array.isArray(cur) ?
            [...pre, ...flatten(cur, depth - 1)] : [...pre, cur], [])
  }
  return flatten(this, depth)
}

参考视频