一天一个仿lodash函数实现-flatten

flatten

数组元素铺平一层

function flatten(arr){
  return arr.reduce((pre, cur)=>{
    return pre.concat(cur)
  }, [])
}

flattenDeep

数组元素全部铺平，采用递归的方式，当遇到元素为数组时，继续调用函数来铺平数组，否则直接concat

function flattenDeep(arr) {
  return arr.reduce((pre, cur)=>{
    return cur instanceof Array? pre.concat(flattenDeep(cur)):pre.concat(cur)
  }, [])
}

flattenDepth

按深度铺平元素, 在deep的基础上增加条件判断

function flattenDepth(arr, depth=1) {
  return arr.reduce((pre, cur)=>{
    return cur instanceof Array && depth>1? pre.concat(flattenDepth(cur, depth - 1)):pre.concat(cur)
  }, [])
}