您的位置:首页 > Web前端 > JavaScript

js 将线性数据转为树形的示例代码

2019-05-28 18:04 1546 查看

在日常开发工作中,我们经常碰到将线性的数据转换成树的需求,今天给大家分享一个简单的转换算法。

数据结构

下面是我们转换前的数据:

[
{
"id":1,
"parent_id":0,
"name":"四川省"
},
{
"id":2,
"parent_id":0,
"name":"广东省"
},
{
"id":3,
"parent_id":0,
"name":"江西省"
},
{
"id":5,
"parent_id":1,
"name":"成都市"
},
{
"id":6,
"parent_id":5,
"name":"锦江区"
},
{
"id":7,
"parent_id":6,
"name":"九眼桥"
},
{
"id":8,
"parent_id":6,
"name":"兰桂坊"
},
{
"id":9,
"parent_id":2,
"name":"东莞市"
},
{
"id":10,
"parent_id":9,
"name":"长安镇"
},
{
"id":11,
"parent_id":3,
"name":"南昌市"
}
]

我们转换后的结果是:

[
{
"id":1,
"parent_id":0,
"name":"四川省",
"children":[
{
"id":5,
"parent_id":1,
"name":"成都市",
"children":[
{
"id":6,
"parent_id":5,
"name":"锦江区",
"children":[
{
"id":7,
"parent_id":6,
"name":"九眼桥"
},
{
"id":8,
"parent_id":6,
"name":"兰桂坊"
}
]
}
]
}
]
},
{
"id":2,
"parent_id":0,
"name":"广东省",
"children":[
{
"id":9,
"parent_id":2,
"name":"东莞市",
"children":[
{
"id":10,
"parent_id":9,
"name":"长安镇"
}
]
}
]
},
{
"id":3,
"parent_id":0,
"name":"江西省",
"children":[
{
"id":11,
"parent_id":3,
"name":"南昌市"
}
]
}
]

实现代码

let array = [
{
id: 1,
parent_id: 0,
name: "四川省"
},
{
id: 2,
parent_id: 0,
name: "广东省"
},
{
id: 3,
parent_id: 0,
name: "江西省"
},
{
id: 5,
parent_id: 1,
name: "成都市"
},
{
id: 6,
parent_id: 5,
name: "锦江区"
},
{
id: 7,
parent_id: 6,
name: "九眼桥"
},
{
id: 8,
parent_id: 6,
name: "兰桂坊"
},
{
id: 9,
parent_id: 2,
name: "东莞市"
},
{
id: 10,
parent_id: 9,
name: "长安镇"
},
{
id: 11,
parent_id: 3,
name: "南昌市"
}
]

function listToTree(list) {
let map = {};
list.forEach(item => {
if (! map[item.id]) {
map[item.id] = item;
}
});

list.forEach(item => {
if (item.parent_id !== 0) {
map[item.parent_id].children ? map[item.parent_id].children.push(item) : map[item.parent_id].children = [item];
}
});

return list.filter(item => {
if (item.parent_id === 0) {
return item;
}
})
}
console.log(listToTree(array));

分析

这段代码的核心就在 listToTree 方法中,这个方法分为了三个部分:

第一部分

第一部分先将数组中的所有元素都复制到 map 中(注意:这里是引用复制哦,这个细节很重要)。

第二部分

执行第二次遍历前的 map:

// map
{
...,
"3":{
"id":3,
"parent_id":0,
"name":"江西省"
},
...
}

然后这个时候遍历 parent_id 不等于 0 的元素:

[
...,
{
id: 11,
parent_id: 3,
name: "南昌市"
},
...
]

然后发现南昌市有 parent_id,我们再给 map[item.parent_id] 设置子元素,通过南昌市的 parent_id 可以推导出:

map["3"].children ? map["3"].children.push(item) : map[3].children = [item];

上面的代码判断了是否存在 children,如果不存在则直接给它赋值,否则将值 push 到 children 中。

执行完第二步后,我们已经把子节点添加到了它的父节点上,但是我们并没有删除掉之前的子节点。所以第三部就是对数据进行过滤,只要父节点即可。

总结

需要注意的是,我们一直都是对 map 进行操作的,但是结果怎么到了 list 上呢,这就是上面提到的引用复制。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

您可能感兴趣的文章:

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: