数组转树(递归法)

let data = [
       { id: '01', name: '张大大', pid: '', job: '项目经理' },
       { id: '02', name: '小亮', pid: '01', job: '产品leader' },
       { id: '03', name: '小美', pid: '01', job: 'UIleader' },
       { id: '04', name: '老马', pid: '01', job: '技术leader' },
       { id: '05', name: '老王', pid: '01', job: '测试leader' },
       { id: '06', name: '老李', pid: '01', job: '运维leader' },
       { id: '07', name: '小丽', pid: '02', job: '产品经理' },
       { id: '08', name: '大光', pid: '02', job: '产品经理' },
       { id: '09', name: '小高', pid: '03', job: 'UI设计师' },
       { id: '10', name: '小刘', pid: '04', job: '前端工程师' },
       { id: '11', name: '小华', pid: '04', job: '后端工程师' },
       { id: '12', name: '小李', pid: '04', job: '后端工程师' },
       { id: '13', name: '小赵', pid: '05', job: '测试工程师' },
       { id: '14', name: '小强', pid: '05', job: '测试工程师' },
       { id: '15', name: '小涛', pid: '06', job: '运维工程师' }
    ]

需求:需要把数组通过pid找到每个元素的上级,并且把这个元素添加到上级的元素中去

做法:

function tree(data,pid=''){
    const arr = []
    data.forEach(item=>{
        if(item.pid===pid){ 
          let children = tree(data,item.id) 
        //    console.log(children);
            if(children){
                item.children = children
            }
            arr.push(item)
        }
    })
    return arr
}
console.log(tree(data));

解析:

function tree(data,pid=''){
    // 传入data数组,和pid进行递归
    
    const arr = []
    // 声明一个新数组接收树形数组
    
    data.forEach(item=>{
        // 遍历数组,找到一级元素,这里的pid为空值是是最上级元素,
        //也就是一级元素,判断找到一级元素
        if(item.pid===pid){ 
            
            // 开始递归,通过一级元素的id值来找到是否有与之对应
            //的二级元素,即一级元素的item.id是否等于二级元素的item.pid
          let children = tree(data,item.id) 
            
            // 接着上面,判断每一个二级元素pid是否跟这个一级元素
            //id值相等,如果相等,则把这个二级元素,添加到一级元素的children属性中   
            if(children){         
                item.children = children
            }

            // 找到一级元素后,把每一个一级元素添加到新数组中,因
            //为是树形结构,所以数组里面的一级数组就是一级元素
            arr.push(item)
        }
    })
    //返回arr新数组的值,此时的arr是一个树形结构
    return arr
}
console.log(tree(data));