let input = [
{
id: 1, val: '学校', parentId: null
}, {
id: 2, val: '班级1', parentId: 1
}, {
id: 3, val: '班级2', parentId: 1
}, {
id: 4, val: '学生1', parentId: 2
}, {
id: 5, val: '学生2', parentId: 3
}, {
id: 6, val: '学生3', parentId: 3
},
]
let output = {
id: 1,
val: '学校',
children: [{
id: 2,
val: '班级1',
children: [
{
id: 4,
val: '学生1',
children: []
},
{
id: 5,
val: '学生2',
children: []
}
]
}, {
id: 3,
val: '班级2',
children: [{
id: 6,
val: '学生3',
children: []
}]
}]
}
function arrayToTree(arr){
const map = {}
let tree = null
arr.forEach((v) => {
map[v.id] = v
map[v.id].children = []
})
arr.forEach((v) => {
if(v.parentId){
map[v.parentId].children.push(v)
}else{
tree = v
}
})
return tree
}
arrayToTree(input)
console