平铺的数组数据转化为树状结构

在做关于有树状构架的时候,我们需要将平铺的数组转化为树状结构,下面我会介绍一种简便的方式,有详细的注释,供大家参考!

export function arrayToTree(list) {
// 1. 定义两个变量
  const treeList = [], map = {}

  // 2. 建立一个映射关系，并给每个元素补充children属性.
  // 映射关系: 目的是让我们能通过id快速找到对应的元素
  // 补充children：让后边的计算更方便
  list.forEach(item => {
    if (!item.children) {
      item.children = []
    }
    map[item.id] = item
  })

  // 循环
  list.forEach(item => {
    // 对于每一个元素来说，先找它的上级
    //    如果能找到，说明它有上级，则要把它添加到上级的children中去
    //    如果找不到，说明它没有上级，直接添加到 treeList
    const parent = map[item.pid]
    // 如果存在上级则表示item不是最顶层的数据
    if (parent) {
      parent.children.push(item)
    } else {
      // 如果不存在上级 则是顶层数据,直接添加
      treeList.push(item)
    }
  })
  // 返回
  return treeList
  // return [{ label: xxx, children: [{label: xxxx, children: [] }]}]
}

使用了两层循环