数组转树的代码详解(小白版)

/**
 * 把平铺的数组结构转成树形结构
 */
const arr = [
  {id: "2b", pid: "", name: "人事部"},
  {id: "2c", pid: "", name: "财务部", },
  {id: "2d", pid: "2c", name: "财务核算部"},
  {id: "2e", pid: "2c", name: "税务管理部"},
  {id: "2f", pid: "2c", name: "薪资管理部"}
    ]

//

要达到的效果如下

 代码如下

export function tranListToTreeData(list) {
  // 1. 定义两个中间变量
  const treeList = [],  // 最终要产出的树状数据的数组
        map = {}        // 存储映射关系
  

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

   // 3. 循环处理每个元素
  list.forEach(item => {
    // 对于每一个元素来说，先找它的上级
    //    如果能找到，说明它有上级，则要把它添加到上级的children中去
    //    如果找不到，说明它没有上级，直接添加到 treeList
    const parent = map[item.pid]
    // map[item.pid]表示在map这个映射对象中寻找上级元素的pid,
    // 如何实现这个功能呢?
    // 假设现在item是第三个元素,那么item.pid = '2c' 
    //  map[item.pid]=map['2c'] 这里的['2c']表示对象map里面的一个索引值,
    // 我们的map里面可以找到这个索引值2c,并且找到其对应的元素 '2c': { 'id': '2c', 'pid': '',     'name': '财务部' },
    // 那么const parent = map[item.pid] 就全等于const parent = '2c': { 'id': '2c', 'pid': '',     'name': '财务部' },
    // parent此时就表示为arr的一个顶层数据,就可以开始下面的判断了

    if (parent) {
        // 如果存在则表示item不是最顶层的数据
        // 向对应的顶层数据的children中添加item数据
      parent.children.push(item)
    } else {
      // 如果不存在 则表示这个item是顶层数据
      // 把这个顶层数据添加到一个新数组中
      treeList.push(item)
    }
  })
  // 返回出去
  return treeList
}
</script>
</html>