巧妙实现扁平数组到嵌套树形结构的转换
本文介绍如何将一个扁平的数组对象，根据其level字段，转换为嵌套的树形结构。 挑战在于数组中对象的层级关系并非直接表达，而是通过level字段隐式表示。 我们需要根据level值判断对象的父子关系，最终构建清晰的嵌套结构。
假设我们有一个如下扁平数组：
const chapters = [{
level: '1',
content: '案例概况010000-170900',
url: '案例概况010000-170900.html'
},{
level: '2',
content: '案例概况010100-170901',
url: '案例概况010100-170901.html'
},{
level: '2',
content: '案例概况010200-170902',
url: '案例概况010200-170902.html'
},{
level: '3',
content: '案例概况010201-170903',
url: '案例概况010201-170903.html'
},{
level: '1',
content: '案例概况020000-170904',
url: '案例概况020000-170904.html'
},{
level: '1',
content: '案例概况030000-170905',
url: '案例概况030000-170905.html'
},{
level: '2',
content: '案例概况030100-170906',
url: '案例概况030100-170906.html'
}];
我们的目标是将其转换为如下嵌套结构：
const tree = [{
level: '1',
content: '案例概况010000-170900',
url: '案例概况010000-170900.html',
children: [{
level: '2',
content: '案例概况010100-170901',
url: '案例概况010100-170901.html',
children: []
},{
level: '2',
content: '案例概况010200-170902',
url: '案例概况010200-170902.html',
children: [{
level: '3',
content: '案例概况010201-170903',
url: '案例概况010201-170903.html',
children: []
}]
}]
},{
level: '1',
content: '案例概况020000-170904',
url: '案例概况020000-170904.html',
children: []
},{
level: '1',
content: '案例概况030000-170905',
url: '案例概况030000-170905.html',
children: [{
level: '2',
content: '案例概况030100-170906',
url: '案例概况030100-170906.html',
children: []
}]
}];
一种高效的解决方案是使用迭代和辅助数组parents跟踪当前处理的父节点。 以下代码展示了实现：
function createTreeByList(arr) {
const tree = [];
const parents = [];
arr.forEach(item => {
item.children = []; // 初始化children属性
while (parents.length > 0 && +parents[parents.length - 1].level >= +item.level) {
parents.pop();
}
if (parents.length === 0) {
tree.push(item);
} else {
parents[parents.length - 1].children.push(item);
}
parents.push(item);
});
return tree;
}
const tree = createTreeByList(chapters);
console.log(JSON.stringify(tree, null, 2));
这段代码利用while循环找到合适的父节点，并将其添加到children属性中。 parents数组确保了父子关系的正确维护。 通过这种迭代方法，我们高效地将扁平数组转换为嵌套树形结构，避免了递归的开销。
