扁平数组转tree结构

扁平数组转tree结构
let arr = [
  { id: 1, name: '部门1', pid: 0 },
  { id: 7, name: '部门1', pid: 0 },
  { id: 2, name: '部门2', pid: 1 },
  { id: 3, name: '部门3', pid: 1 },
  { id: 4, name: '部门4', pid: 3 },
  { id: 5, name: '部门5', pid: 4 },
]
const list = []
const ato = {}
arr.map((item) => {
  item.children = []
  ato[item.id] = item
  if (item.pid == 0) list.push(item)
})
arr.map((item) => {
  if (ato[item.pid]) ato[item.pid].children.push(item)
})
console.log(list)