js將列表組裝成樹結構的兩種實現方式分享

更新時間：2022年01月09日 11:29:41 作者：Orime小豬

最近做的任務提了新的需求,需要實現一個樹形結構,所以下面這篇文章主要給大家介紹了關于js將列表組裝成樹結構的兩種實現方式,需要的朋友可以參考下

前言

工作中偶爾就會遇到后端同學丟來一個列表，要我們自己組裝成一個樹結構渲染到頁面上，本文以兩種不同方式探索生成樹的算法思想。

背景介紹

可組裝成樹結構的數組一般有以下幾個要素：

當前節(jié)點id
parentId 當前節(jié)點的父節(jié)點id
children 子節(jié)點列表（可能不會在接口中返回，需要組裝時候自己加上）

原始結構：

目標結構：

關鍵就是一維數組中通過parentId找到其對應的父節(jié)點并添加到父節(jié)點的children數組中。

實現方案

最直接的方式就是遍歷數組，并把找到的子節(jié)點逐一添加到父節(jié)點中

function listToTreeSimple(data) {
  const res = [];
  data.forEach((item) => {
    const parent = data.find((node) => node.id === item.parentId);
    if (parent) {
      parent.children = parent.children || [];
      parent.children.push(item);
    } else {
      // * 根節(jié)點
      res.push(item);
    }
  });
  return res;
}

考慮進一步優(yōu)化，使用哈希表，以id為key存儲每個節(jié)點值，省去data.find計算

function listToTree(data) {
  // * 先生成parent建立父子關系
  const obj = {};
  data.forEach((item) => {
    obj[item.id] = item;
  });
  // * obj -> {1001: {id: 1001, parentId: 0, name: 'AA'}, 1002: {...}}
  // console.log(obj, "obj")
  const parentList = [];
  data.forEach((item) => {
    const parent = obj[item.parentId];
    if (parent) {
      // * 當前項有父節(jié)點
      parent.children = parent.children || [];
      parent.children.push(item);
    } else {
      // * 當前項沒有父節(jié)點 -> 頂層
      parentList.push(item);
    }
  });
  return parentList;
}

即便數據量很小，帶來的性能提升也是顯著的

遞歸法

更有騷操作遞歸法，性能會很差，但代碼會很酷??

function recursiveToTree(data) {
  function loop(key) {
    const arr = [];
    data.forEach((item) => {
      if (item.parentId === key) {
        item.children = loop(item.id);
        arr.push(item);
      }
    });
    return res;
  }
  return loop(1);
}

看看性能，誒？看起來竟然遞歸性能最佳??

但是數據量稍微大一點……

(上面遞歸，下面非遞歸)

資源

原始數據列表

const list = [
  {
    id: 1001,
    parentId: 0,
    name: "AA",
  },
  {
    id: 1002,
    parentId: 1001,
    name: "BB",
  },
  {
    id: 1003,
    parentId: 1001,
    name: "CC",
  },
  {
    id: 1004,
    parentId: 1003,
    name: "DD",
  },
  {
    id: 1005,
    parentId: 1003,
    name: "EE",
  },
  {
    id: 1006,
    parentId: 1002,
    name: "FF",
  },
  {
    id: 1007,
    parentId: 1002,
    name: "GG",
  },
  {
    id: 1008,
    parentId: 1004,
    name: "HH",
  },
  {
    id: 1009,
    parentId: 1005,
    name: "II",
  },
];