本文详细解析了 JavaScript 中树形结构与数组结构的相互转换方法,以及如何在树形数据中高效查找特定节点。内容涵盖深度优先与广度优先遍历算法实现,以及扁平化树形数据的两种思路,并介绍了将数组还原为树形结构的优化写法,适用于前端开发中的菜单、权限管理等场景。
在实际开发中,处理树形结构的需求十分常见,因此有必要总结几种常用操作的实现方式。
首先,假设存在如下所示的树形结构数据:
var tree=[
{
'id': '1',
'name': '教学素材管理',
'children':[
{
'id': '101',
'name': '教学素材',
'children':[
{
'id': '10101',
'name': '修改',
},
{
'id': '10102',
'name': '添加',
}
]
}, {
'id': '102',
'name': '测试试题',
},
{
'id': '103',
'name': '问题任务',
}
]
}, {
'id': '2',
'name': '基础数据管理',
'children':[
{
'id': '201',
'name': '专业设置',
},
{
'id': '202',
'name': '专业管理',
}
]
}
]1、如何在 tree 中找到 id=10102 的对象?
思路一:采用深度遍历法。从根节点开始,若当前节点包含子节点,则递归遍历其子节点。
function deepQuery(tree,id) {
var isGet = false;
var retNode = null;
function deepSearch(tree,id){
for(var i = 0; i<tree.length; i++) {
if(tree[i].children && tree[i].children.length>0) {
deepSearch(tree[i].children,id);
}
if(id === tree[i].id || isGet) {
isGet||(retNode = tree[i]);
isGet = true;
break;
}
}
}
deepSearch(tree,id);
return retNode;
}调用示例:
var getNode = deepQuery(tree,10102);
console.log(getNode)思路二:采用广度遍历法。依次遍历根节点的所有直接子节点,随后从第一个子节点开始继续向下遍历。
function breadthQuery(tree, id) {
var stark = [];
stark = stark.concat(tree);
while(stark.length) {
var temp = stark.shift();
if(temp.children) {
stark = stark.concat(temp.children);
}
if(temp.id === id) {
return temp;
}
}
}调用示例:
var getNode=breadthQuery(tree,10102);
console.log(getNode);2、如何将树形结构转换为具有父子关系属性的数组结构?
思路一:初始化一个空数组,从树形结构的顶层开始遍历。当当前节点包含子节点时,在继续遍历子节点的同时,从当前节点中移除子节点属性,并将当前节点对象推入空数组。
function flatten1(tree) {
var arr = [];
function spread(tree,pid) {
for (var i=0; i < tree.length; i++ ) {
item = tree[i]
let {id,name}=item;
arr.push({id,name,pid})
if (item.children) {
spread(item.children,item.id)
delete item.children
}
}
}
spread(tree,0)
return arr;
}
var newArr = flatten1(tree);思路二:
function flatten2 (data,pid) {
return data.reduce((arr, {id, name, children = []}) =>
arr.concat([{id, name,pid}], flatten2(children,id)), [])
}
var newArr = flatten2(tree,0);转换结果:

3、如何将数组结构转换为树形结构?
以下是参考某位开发者分享的简洁写法(原文链接见注):
function treeData(data){
let cloneData = JSON.parse(JSON.stringify(data))
return cloneData.filter(parent=>{
let branchArr = cloneData.filter(child => parent['id'] == child['pid']);
branchArr.length>0 ? parent['children'] = branchArr : '';
return parent['pid'] == 0 ;
})
}调用示例:
var newTree = treeData(newArr)使用此前测试生成的数组数据进行验证

结果如上所示,成功还原了最初的树形结构。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。