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));