在日常開發工作中,我們經常碰到將線性的數據轉換成樹的需求,今天給大家分享一個簡單的轉換算法。
數據結構
下面是我們轉換前的數據:
[ { "id":1, "parent_id":0, "name":"四川省" }, { "id":2, "parent_id":0, "name":"廣東省" }, { "id":3, "parent_id":0, "name":"江西省" }, { "id":5, "parent_id":1, "name":"成都市" }, { "id":6, "parent_id":5, "name":"錦江區" }, { "id":7, "parent_id":6, "name":"九眼橋" }, { "id":8, "parent_id":6, "name":"蘭桂坊" }, { "id":9, "parent_id":2, "name":"東莞市" }, { "id":10, "parent_id":9, "name":"長安鎮" }, { "id":11, "parent_id":3, "name":"南昌市" }]
我們轉換后的結果是:
[ { "id":1, "parent_id":0, "name":"四川省", "children":[ { "id":5, "parent_id":1, "name":"成都市", "children":[ { "id":6, "parent_id":5, "name":"錦江區", "children":[ { "id":7, "parent_id":6, "name":"九眼橋" }, { "id":8, "parent_id":6, "name":"蘭桂坊" } ] } ] } ] }, { "id":2, "parent_id":0, "name":"廣東省", "children":[ { "id":9, "parent_id":2, "name":"東莞市", "children":[ { "id":10, "parent_id":9, "name":"長安鎮" } ] } ] }, { "id":3, "parent_id":0, "name":"江西省", "children":[ { "id":11, "parent_id":3, "name":"南昌市" } ] }]
實現代碼
let array = [ { id: 1, parent_id: 0, name: "四川省" }, { id: 2, parent_id: 0, name: "廣東省" }, { id: 3, parent_id: 0, name: "江西省" }, { id: 5, parent_id: 1, name: "成都市" }, { id: 6, parent_id: 5, name: "錦江區" }, { id: 7, parent_id: 6, name: "九眼橋" }, { id: 8, parent_id: 6, name: "蘭桂坊" }, { id: 9, parent_id: 2, name: "東莞市" }, { id: 10, parent_id: 9, name: "長安鎮" }, { id: 11, parent_id: 3, name: "南昌市" }]function listToTree(list) { let map = {}; list.forEach(item => { if (! map[item.id]) { map[item.id] = item; } }); list.forEach(item => { if (item.parent_id !== 0) { map[item.parent_id].children ? map[item.parent_id].children.push(item) : map[item.parent_id].children = [item]; } }); return list.filter(item => { if (item.parent_id === 0) { return item; } })}console.log(listToTree(array));
分析
這段代碼的核心就在 listToTree 方法中,這個方法分為了三個部分:
第一部分
第一部分先將數組中的所有元素都復制到 map 中(注意:這里是引用復制哦,這個細節很重要)。
第二部分
執行第二次遍歷前的 map:
// map{ ..., "3":{ "id":3, "parent_id":0, "name":"江西省" }, ...}
然后這個時候遍歷 parent_id 不等于 0 的元素:
[ ..., { id: 11, parent_id: 3, name: "南昌市" }, ...]
然后發現南昌市有 parent_id,我們再給 map[item.parent_id] 設置子元素,通過南昌市的 parent_id 可以推導出:
map["3"].children ? map["3"].children.push(item) : map[3].children = [item];
上面的代碼判斷了是否存在 children,如果不存在則直接給它賦值,否則將值 push 到 children 中。
執行完第二步后,我們已經把子節點添加到了它的父節點上,但是我們并沒有刪除掉之前的子節點。所以第三部就是對數據進行過濾,只要父節點即可。
總結
需要注意的是,我們一直都是對 map 進行操作的,但是結果怎么到了 list 上呢,這就是上面提到的引用復制。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持VeVb武林網。
新聞熱點
疑難解答