使用generator压平一个数组

function* iter(tree) {
    if(Array.isArray(tree)) {
        for(let v of tree) {
            yield* flat(v)
        }
    }else {
        yield tree
    }
}

function flat(arr) {
  return [...iter(arr)]
}
const arr = [1, [2,[3],4], 5]
const arr2 = flat(arr)
console.log(arr2)