第一种方法:使用map映射
const arr = [
{ id: 1, name: '部门A', parentId: 0 },
// 0 代表顶级节点,无父节点
{ id: 2, name: '部门B', parentId: 1 },
{ id: 3, name: '部门C', parentId: 1 },
{ id: 4, name: '部门D', parentId: 2 },
{ id: 5, name: '部门E', parentId: 2 },
{ id: 6, name: '部门F', parentId: 3 }
]
function arrTtree(arr){
const treeList = new Map()
let root = null
arr.forEach( item => {
const { id, name, parentId } = item
// 定义 tree node 并加入 map
const treeNode = { id, name }
treeList.set(id, treeNode)
// 找到 parentNode 并加入到它的 children
const parentNode = treeList.get(parentId)
if (parentNode) {
if (parentNode.children == null) parentNode.children = []
parentNode.children.push(treeNode)
}
// 找到根节点
if (parentId === 0) root = treeNode
})
return root
}
控制台输出如下:
第二种方法:累加+递归
const arr = [
{ id: 1, name: '部门A', parentId: 0 },
// 0 代表顶级节点,无父节点
{ id: 2, name: '部门B', parentId: 1 },
{ id: 3, name: '部门C', parentId: 1 },
{ id: 4, name: '部门D', parentId: 2 },
{ id: 5, name: '部门E', parentId: 2 },
{ id: 6, name: '部门F', parentId: 3 }
]
function arrTtree(parentId) {
return arr.reduce( (prev,current) => {
if(current.parendId === parentId) {
current.children = arrTree(current.id)
prev.push(current)
}
return prev
},[])
}
console.log(arrTtree(0))
控制台输出如下: