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

更新時(shí)間：2019年05月28日 09:45:12 作者：sizeofio

這篇文章主要介紹了js 將線性數(shù)據(jù)轉(zhuǎn)為樹形的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

在日常開發(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)即可。