有序且无父id数组转树结构

最近业务上遇到一个数组转树结构的问题,觉得有些意思,决定写一写算法 

提出问题

一个数组表示树,level表示其层级,每一个节点都是它前边最近一个小于其level的节点的子级,怎样把这个数组转化为树,如下,怎么图1 中的数组转换为图2中的数结构

图1

[
  { name: "A", level: 0 },
  { name: "B", level: 0 },
  { name: "b1", level: 1 },
  { name: "b2", level: 1 },
  { name: "b2-1", level: 2 },
  { name: "b2-2", level: 2 },
  { name: "b2-3", level: 2 },
  { name: "b3", level: 1 },
  { name: "C", level: 0 },
]

图2 

[
  {
    "name": "A",
    "level": 0
  },
  {
    "name": "B",
    "level": 0,
    "children": [
      {
        "name": "b1",
        "level": 1
      },
      {
        "name": "b2",
        "level": 1,
        "children": [
          {
            "name": "b2-1",
            "level": 2
          },
          {
            "name": "b2-2",
            "level": 2
          },
          {
            "name": "b2-3",
            "level": 2
          }
        ]
      },
      {
        "name": "b3",
        "level": 1
      }
    ]
  },
  {
    "name": "C",
    "level": 0
  }
]

分析问题

可以创建一个栈,遍历当前数组,当数组中数据的level小于栈顶数据的level时,将栈中数据弹出,将弹出的数据组成数组,加入到栈顶数据的children中,如下图

 

 

 

解决问题

废话少说,上代码

function listToTree(list) {
  let arr = [];
  for (let i = 0; i < list.length; i++) {
    let item = list[i];
    if (arr.length < 1 || arr[arr.length -1].level <= item.level) {
      arr.push(item)
    } else {
      let temp = []
      while(arr && arr.length > 0 && arr[arr.length -1].level > item.level) {
        temp.unshift(arr.pop())
      }
      arr[arr.length - 1].children = temp;
      arr.push(item);
    }
  }
  return arr;
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值