最近业务上遇到一个数组转树结构的问题,觉得有些意思,决定写一写算法
提出问题
一个数组表示树,level表示其层级,每一个节点都是它前边最近一个小于其level的节点的子级,怎样把这个数组转化为树,如下,怎么图1 中的数组转换为图2中的数结构
图1
[
{ name: "A", level: 0 },
{ name: "B", level: 0 },
{ name: "b1", level: 1 },
{ name: "b2", level: 1 },
{ name: "b2-1", level: 2 },
{ name: "b2-2", level: 2 },
{ name: "b2-3", level: 2 },
{ name: "b3", level: 1 },
{ name: "C", level: 0 },
]
图2
[
{
"name": "A",
"level": 0
},
{
"name": "B",
"level": 0,
"children": [
{
"name": "b1",
"level": 1
},
{
"name": "b2",
"level": 1,
"children": [
{
"name": "b2-1",
"level": 2
},
{
"name": "b2-2",
"level": 2
},
{
"name": "b2-3",
"level": 2
}
]
},
{
"name": "b3",
"level": 1
}
]
},
{
"name": "C",
"level": 0
}
]
分析问题
可以创建一个栈,遍历当前数组,当数组中数据的level小于栈顶数据的level时,将栈中数据弹出,将弹出的数据组成数组,加入到栈顶数据的children中,如下图
解决问题
废话少说,上代码
function listToTree(list) {
let arr = [];
for (let i = 0; i < list.length; i++) {
let item = list[i];
if (arr.length < 1 || arr[arr.length -1].level <= item.level) {
arr.push(item)
} else {
let temp = []
while(arr && arr.length > 0 && arr[arr.length -1].level > item.level) {
temp.unshift(arr.pop())
}
arr[arr.length - 1].children = temp;
arr.push(item);
}
}
return arr;
}