一 ,为什么需要数组转树:
当我们做后台管理系统时难免会遇到关于公司组织架构这样的模块,一个部门下会有好几个小部门,这时我们就可以运用树形图来更好地进行查看,下面简单举例。
展开前:
展开后:
二,下面使用两种方法进行数组转树:
1. 非递归方式:
/**
* 把平铺的数组结构转成树形结构
*
* [
* { 'id': '29', 'pid': '', 'name': '总裁办' },
{ 'id': '2c', 'pid': '', 'name': '财务部' },
{ 'id': '2d', 'pid': '2c', 'name': '财务核算部'},
{ 'id': '2f', 'pid': '2c', 'name': '薪资管理部'},
{ 'id': 'd2', 'pid': '', 'name': '技术部'},
{ 'id': 'd3', 'pid': 'd2', 'name': 'Java研发部'}
* ]
* 上面的结构说明: pid为空的为一级部门, pid等于id的为当前部门的二级部门
*/
export function tranListToTreeData(list) {
// 1. 定义两个变量
const treeList = []; const map = {}
// 2. 建立一个映射关系,并给每个元素补充children属性.
// 映射关系: 目的是让我们能通过id快速找到对应的元素
// 补充children:让后边的计算更方便
list.forEach(item => {
if (!item.children) {
item.children = []
}
map[item.id] = item
})
// 循环
list.forEach(item => {
// 对于每一个元素来说,先找它的上级
// 如果能找到,说明它有上级,则要把它添加到上级的children中去
// 如果找不到,说明它没有上级,直接添加到 treeList
const parent = map[item.pid]
// 如果存在上级则表示item不是最顶层的数据
if (parent) {
parent.children.push(item)
} else {
// 如果不存在上级 则是顶层数据,直接添加
treeList.push(item)
}
})
// 返回
return treeList
}
非递归方式代码量稍微多一点
2. 递归方式
/* let arr=[
* { 'id': '29', 'pid': '', 'name': '总裁办' },
* { 'id': '2c', 'pid': '', 'name': '财务部' },
* .......
* ]
*/
function tranListToTreeData(list,pid='') {
// 先找到所有的根节点
let tranList=list.filter(it=>it.pid===pid)
tranList.forEach(itm =>{
//传入id list进行递归 在筛选出 他的父级 是一个数组
itm.children=tranListToTreeData(list,itm.id)
})
return tranList
}
console.log(tranListToTreeData(arr));