let arr = [{
id: 1,
name: '部門1',
pid: 0
},
{
id: 2,
name: '部門2',
pid: 1
},
{
id: 3,
name: '部門3',
pid: 1
},
{
id: 4,
name: '部門4',
pid: 3
},
{
id: 5,
name: '部門5',
pid: 4
},
]
如上一個(gè)數(shù)據(jù)結(jié)構(gòu),將他轉(zhuǎn)為樹形結(jié)構(gòu)。
方法一:
function queryChilrden(arr) {
let result = []
arr.forEach(s => {
s.chilrden=[]
arr.forEach(d=>{
if(s.id==d.pid){
s.chilrden.push(d)
}
})
})
return arr.find(item => item.pid === 0)
}
解析:
遍歷兩次原數(shù)組,并在第一次遍歷的時(shí)候給數(shù)組的每一項(xiàng)添加一個(gè)chilrden屬性。在第二次遍歷的時(shí)候就開始比較,用本身id去比對(duì)每一項(xiàng)的pid。如果相等,即說明找到了兒子,那就將其push進(jìn)自身的chilrden數(shù)組里。最后return的時(shí)候做個(gè)篩選,返回最高等級(jí)的也就是pid為0的那一項(xiàng)。
方法二:
function mapFun(arr){
let mapItem = {}
let result = []
arr.forEach(d=>{
mapItem[d.id] ={...d,chilrden:[]}
})
arr.forEach(s=>{
if(!s.pid){
result.push(mapItem[s.id])
}else{
if(mapItem[s.pid]){
//用s.pid為key去mapItem里找,如果找到了。就說明當(dāng)前這一項(xiàng)在mapItem里有父親,然后就用父親將當(dāng)前s push進(jìn)去。
mapItem[s.pid].chilrden.push(mapItem[s.id])
}
}
})
return result
}
解析:第二種方法其實(shí)很第一種方法雷同,但是第二種方法用到了map結(jié)構(gòu)。首先準(zhǔn)備一個(gè)mapItem的對(duì)象,用來存放所有數(shù)據(jù)。
第一次遍歷,將數(shù)組里的數(shù)據(jù)轉(zhuǎn)成map結(jié)構(gòu)存入mapItem對(duì)象中。其實(shí)就是將每一項(xiàng)的id作為key,這樣會(huì)方便后面查找數(shù)據(jù)。提高效率
第二次遍歷的時(shí)候就要開始找數(shù)據(jù)了。首先把pid為0的拎出來,找到元素之后要注意,這里不再是存儲(chǔ)遍歷中的數(shù)據(jù)。而是通過遍歷中的數(shù)據(jù)去找對(duì)應(yīng)的map數(shù)據(jù),將map數(shù)據(jù)存起來。
可以理解為mapItem就是一本字典。比如我們要找pid為0的那一項(xiàng)。找到了之后,在mapItem這本字典中用當(dāng)前項(xiàng)的id作為key,就可以找到這個(gè)元素
要記住存的時(shí)候要用對(duì)象的id去mapItem里找,而不是用對(duì)象的pid去找。因?yàn)槲覀冊(cè)诖鎚apItem的時(shí)候用的是對(duì)象的id為key
if(mapItem[s.pid]){
mapItem[s.pid].chilrden.push(mapItem[s.id])
}
在mapItem中key是每一項(xiàng)的id。已知,元素的pid等于父親的id。所以這里用mapItem[s.pid]去找父親。如果存在,那就說明當(dāng)前遍歷到的這一個(gè)元素,有父親。所以就將當(dāng)前元素push進(jìn)去。push的時(shí)候注意,不再是用pid去找元素了,而是用id了。因?yàn)槲覀僷ush是要push元素本身,而不是再去找他父親
方法三:
// 遞歸
const toTree = (items, id = null, link = 'parentId') =>
items
.filter(item => item[link] == id)
.map(item => ({
...item,
children: toTree(items, item.id)
}));
解析:通過數(shù)組的filter方法過濾出所有的父級(jí),然后遍歷父級(jí)這個(gè)數(shù)組,讓他的children屬性繼續(xù)去遞歸調(diào)用這個(gè)函數(shù)。