这个很简单的题目在三次笔试,一次面试中遇到了,常理来说应该非常熟练了,但是最后面试的时候由于紧张还是只想到了大致思路,没写出来,非常遗憾,现整理一下思路。
let arr=[
{id:1,name:'l1'},
{id:2,name:'l2',parentId:1},
{id:4,name:'l4',parentId:3},
{id:3,name:'l3',parentId:2},
{id:8,name:'l8',parentId:7}
]
数组中的每一个对象都有其id和parentId,要求最终返回的是一个根节点,该根节点对象要有id,name,children三个属性,children数组中的每一个元素又是一个同样也有children属性的对象,这样构成的嵌套就能根据对象地址之间的指向关系形成树形结构。
所以需要给每一个对象添加一个childen属性。
先根据id遍历一遍,将每一个对象添加到map中,key就是id,value为添加了children属性的对象。
再根据parentId遍历一遍,将每一个对象添加到其父元素的children数组中。
代码如下:
function createTree(arr){
let map=new Map();
let len=arr.length;
let id;
for(let i=0;i<len;i++){
if(!map.has(arr[i].id)){
arr[i].children=[];
map.set(arr[i].id,arr[i]);
}
//根节点
if(arr[i].parentId==undefined)
id=arr[i].id;
}
for(let i=0;i<len;i++){
if(map.has(arr[i].parentId)){
let tmp=map.get(arr[i].parentId);
tmp.children.push(arr[i]);
}
}
return map.get(id);
}