js中平铺数组数据处理为需要的树形数据

很实用两个方法,拿去用吧

方法一

 /**
 * 把平铺的数组结构转成树形结构
 *
 * [
 *  {id:"01", pid:"",   "name":"老王" },
 *  {id:"02", pid:"01", "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
  })

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

方法二

 /**
 * 把平铺的数组结构转成树形结构
 */
// [
//   { 'id': '29', 'pid': '',     'name': '总裁办' },
//   { 'id': '2c', 'pid': '',     'name': '财务部' },
//   { 'id': '2d', 'pid': '2c', 'name': '财务核算部'},
//   { 'id': '2f', 'pid': '2c', 'name': '薪资管理部'},
//   { 'id': 'd2', 'pid': '',     'name': '技术部'},
//   { 'id': 'd3', 'pid': 'd2', 'name': 'Java研发部'}
// ]

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

  // 2. 建立一个映射关系，并给每个元素补充children属性.
  // 映射关系: 目的是让我们能通过id快速找到对应的元素
  // 补充children：让后边的计算更方便
  
  list.forEach(item => {
    if (!item.children) {
      item.children = []
    }
    map[item.id] = item
    //map=== {
      // 29:  { 'id': '29', 'pid': '',     'name': '总裁办' },
      // 2c:  { 'id': '2c', 'pid': '',     'name': '财务部' },
      // 2d:  { 'id': '2d', 'pid': '2c', 'name': '财务核算部'},
      // 2f:  { 'id': '2f', 'pid': '2c', 'name': '薪资管理部'},
      // d2:  { 'id': 'd2', 'pid': '',     'name': '技术部'},
      // d3:  { 'id': 'd3', 'pid': 'd2', 'name': 'Java研发部'}
    // }
  })


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