js 將線性數(shù)據(jù)轉(zhuǎn)為樹形的示例代碼

2019-11-19 11:27:09

字體：大中小

供稿：網(wǎng)友

在日常開(kāi)發(fā)工作中，我們經(jīng)常碰到將線性的數(shù)據(jù)轉(zhuǎn)換成樹的需求，今天給大家分享一個(gè)簡(jiǎn)單的轉(zhuǎn)換算法。

數(shù)據(jù)結(jié)構(gòu)

下面是我們轉(zhuǎn)換前的數(shù)據(jù)：

[  {    "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":"錦江區(qū)"  },  {    "id":7,    "parent_id":6,    "name":"九眼橋"  },  {    "id":8,    "parent_id":6,    "name":"蘭桂坊"  },  {    "id":9,    "parent_id":2,    "name":"東莞市"  },  {    "id":10,    "parent_id":9,    "name":"長(zhǎng)安鎮(zhèn)"  },  {    "id":11,    "parent_id":3,    "name":"南昌市"  }]

我們轉(zhuǎn)換后的結(jié)果是：

[  {    "id":1,    "parent_id":0,    "name":"四川省",    "children":[      {        "id":5,        "parent_id":1,        "name":"成都市",        "children":[          {            "id":6,            "parent_id":5,            "name":"錦江區(qū)",            "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":"長(zhǎng)安鎮(zhèn)"          }        ]      }    ]  },  {    "id":3,    "parent_id":0,    "name":"江西省",    "children":[      {        "id":11,        "parent_id":3,        "name":"南昌市"      }    ]  }]

實(shí)現(xiàn)代碼

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: "錦江區(qū)"  },  {    id: 7,    parent_id: 6,    name: "九眼橋"  },  {    id: 8,    parent_id: 6,    name: "蘭桂坊"  },  {    id: 9,    parent_id: 2,    name: "東莞市"  },  {    id: 10,    parent_id: 9,    name: "長(zhǎng)安鎮(zhèn)"  },  {    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 方法中，這個(gè)方法分為了三個(gè)部分：

第一部分

第一部分先將數(shù)組中的所有元素都復(fù)制到 map 中(注意：這里是引用復(fù)制哦，這個(gè)細(xì)節(jié)很重要)。

第二部分

執(zhí)行第二次遍歷前的 map:

// map{ ..., "3":{    "id":3,    "parent_id":0,    "name":"江西省"  },  ...}

然后這個(gè)時(shí)候遍歷 parent_id 不等于 0 的元素：

[ ..., { id: 11, parent_id: 3, name: "南昌市" },  ...]

然后發(fā)現(xiàn)南昌市有 parent_id，我們?cè)俳o map[item.parent_id] 設(shè)置子元素，通過(guò)南昌市的 parent_id 可以推導(dǎo)出：

map["3"].children ? map["3"].children.push(item) : map[3].children = [item];

上面的代碼判斷了是否存在 children，如果不存在則直接給它賦值，否則將值 push 到 children 中。

執(zhí)行完第二步后，我們已經(jīng)把子節(jié)點(diǎn)添加到了它的父節(jié)點(diǎn)上，但是我們并沒(méi)有刪除掉之前的子節(jié)點(diǎn)。所以第三部就是對(duì)數(shù)據(jù)進(jìn)行過(guò)濾，只要父節(jié)點(diǎn)即可。