项目场景:
最近工作上遇到一个需要把普通数组转化为树的情况
问题描述
项目使用kendoUI,使用到了他的treeView组件,格式上需要树结构,但是后端数据是普通数组。
两种解决思路一种是利用map存储,一种就是单纯的遍历寻找parentId===id,不过需要双重循环,算法复杂度增高不推荐
解决方案:
直接上代码:
const arr = [
{ id: 1, name: '爷爷1', parentId: 0 },
{ id: 2, name: '爸爸2', parentId: 1 },
{ id: 3, name: '爸爸3', parentId: 1 },
{ id: 4, name: '儿子4', parentId: 2 },
{ id: 5, name: '儿子5', parentId: 3 },
{ id: 6, name: '儿子6', parentId: 3 },
]
function change(arr){
let result= []
const nodeMap = new Map()
arr.forEach(element => {
const { id ,name ,parentId } = element
const node ={ id ,name }
nodeMap.set( id, node)
const parentNode =nodeMap.get(parentId)
if(parentId===0) result.push(node)
if(parentId){
if(parentNode.children==null) parentNode.children=[]
parentNode.children.push(node)
}
});
return result
}
const tree=change(arr)
console.log(tree)
另一种:
function change2(arr){
let result=[]
arr.forEach((element,index) => {
if(element.parentId===0){
result.push(element)
}
if(element.parentId!==0){
let id = element.parentId;
arr.forEach((item,index) =>{
if(id===item.id) {
if(arr[index].children==null) arr[index].children=[]
arr[index].children.push(element)
}
})
}
});
return result
}```
扩展:数组是乱序怎么办?
先排序再遍历
```javascript
function sortArr(attr) {
return function(a,b) {
return a[attr] - b[attr]
}
}
arr.sort(sortArr('parentId'));