数组转成树

390 阅读1分钟

普通方法:

// 数组转换为树形的函数

export function tranListToTreeData(list) {
  // 最终要产出的树状数据的数组
  const treeList = []
  // 所有项都使用对象存储起来
  const map = {}

  // 建立一个映射关系:通过id快速找到对应的元素
  list.forEach(item => {
    if (!item.children) {
      item.children = []
    }
    map[item.id] = item
  })
  // map:
  // {
  //   "312c": { 'id': '312c', 'pid': '',     'name': '财务部',    children: [{ 'id': '312d', 'pid': '312c', 'name': '财务核算部',children: []}] },
  //   "312d": { 'id': '312d', 'pid': '312c', 'name': '财务核算部',children: []}
  // }

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

递归:

    let arr=[
        { '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研发部'}
    ]
    function tranListToTreeData(list,pid='') {
        // 先找到所有的根节点
        let tranList=list.filter(it=>it.pid===pid)
        tranList.forEach(itm =>{
            //传入id  list进行递归 在筛选出 他的父级 是一个数组
            itm.children=tranListToTreeData(list,itm.id)
        })
        return tranList
    }

    console.log(tranListToTreeData(arr));