線性結構與樹形結構相互轉換(ES6實現)

前言

  當樹形結構的層級愈來愈深時,操做某一節點會變得愈來愈費勁,維護成本不斷增長。因此線性結構與樹形的相互轉換變得異常重要!node

  首先,咱們約定樹形結構以下:數組

node = {
  id: number,  // 數值
parentId: number, // 數值 name: string,
children: [] || null, // 用數組的方式保存子節點,適合更多業務場景 }

   線性結構:spa

list = [
   { id: number, parentId: number, name: string }, 
   { id: number, parentId: number, name: string },
];

特殊狀況  

  上面的樹形結構並非很完美,當遇到菜單或者分類等業務場景時,每一個頂級節點的parentId約定爲0,當存在多個頂級節點,顯得不是一個完整的樹。因此在這類特殊狀況下,咱們須要構造一個頂級節點。將菜單或者分類的原有頂級節點存儲至該節點的children中。 因此最後約定頂級節點以下。code

root = null || {
  id: 0,
  parentId: null,
  children: [node1, node2, ...],  
}

線性結構與樹形結構相互轉換

  線性轉樹形:blog

function listConvertTree(list) {
  let root = null;
  if (list && list.length) {
    root = { id: 0, parentId: null, children: [] };
    const group = {};
    for (let index = 0; index < list.length; index += 1) {
      if (list[index].parentId !== null && list[index].parentId !== undefined) {
        if (!group[list[index].parentId]) {
          group[list[index].parentId] = [];
        }
        group[list[index].parentId].push(list[index]);
      }
    }
    const queue = [];
    queue.push(root);
    while (queue.length) {
      const node = queue.shift();
      node.children = group[node.id] && group[node.id].length ? group[node.id] : null;
      if (node.children) {
        queue.push(...node.children);
      }
    }
  }
  return root;
}

  樹形轉線性:string

function treeConvertList(root) {
  const list = [];
  if (root) {
    const Root = JSON.parse(JSON.stringify(root));
    const queue = [];
    queue.push(Root);
    while (queue.length) {
      const node = queue.shift();
      if (node.children && node.children.length) {
        queue.push(...node.children);
      }
      delete node.children;
      if (node.parentId !== null && node.parentId !== undefined) {
        list.push(node);
      }
    }
  }
  return list;
}
相關文章
相關標籤/搜索