一组数组结构的数据,包含id以及pid(parentId),根据给出的数据把数组转换成树形结构数据
var flatLs = [{
id: 1
}, {
id: 2
}, {
id: 3
}, {
id: 4,
pid: 1
}, {
id: 5,
pid: 3
}, {
id: 6,
pid: 2
}, {
id: 7,
pid: 4
}, {
id: 8,
pid: 3
}, {
id: 9,
pid: 4
}]
const flat2tree = flatLs => {
flatLs.forEach(item => {
if (!item.pid) {
item.pid = 0;
} else {
const index = flatLs.findIndex(_item => _item.id === item.pid)
if (index !== -1) {
flatLs[index].children = flatLs[index].children || []
flatLs[index].children.push(item)
}
}
})
return flatLs.filter(item => item.pid === 0)
}
console.log(flat2tree(flatLs))
这里主要利用了对象是通过引用传值的特点来处理的。因此遍历整个数组,直接修改传入的参数,最后取出所有的顶层节点的数据即可。数组中无pId的数据可以令pId为0,当做顶层数据。