一、使用递归
// 转树形
const dataTree = (data, pid) => {
const res = []
data.forEach((item) => {
if (item.sjid === pid) {
const itemChildren = dataTree(data, item.id)
if (itemChildren.length) item.children = itemChildren
res.push(item)
}
})
return res
}
二、树形拍平
// 树形拍平
const getNode = (treeData) => {
const flatData = []
function flatten(node, parent) {
const flatNode = { ...node }
flatNode.parentId = parent
flatData.push(flatNode)
if (node.children) {
node.children.forEach((child) => {
flatten(child, node.id)
})
delete flatNode.children
}
}
treeData.forEach((node) => {
flatten(node, null)
})
// console.log(flatData)
return flatData
}
三、通过循环map映射
/**
* 把平铺的数组结构转成树形结构
*/
export function tranListToTreeData(list) {
// 1. 定义两个变量
const treeList = []
const map = {}
// 2. 建立一个映射关系,并给每个元素补充children属性.
// 映射关系: 目的是让我们能通过id快速找到对应的元素
// 补充children:让后边的计算更方便
list.forEach(item => {
if (!item.children) {
item.children = []
}
map[item.xh] = item
})
// 循环
list.forEach(item => {
// 对于每一个元素来说,先找它的上级
// 如果能找到,说明它有上级,则要把它添加到上级的children中去
// 如果找不到,说明它没有上级,直接添加到 treeList
const parent = map[item.pid]
// 如果存在上级则表示item不是最顶层的数据
if (parent) {
parent.children.push(item)
} else {
// 如果不存在上级 则是顶层数据,直接添加
treeList.push(item)
}
})
// 返回
return treeList
}