如何高效地将扁平数组转换为嵌套树形结构？

巧妙实现扁平数组到嵌套树形结构的转换

本文介绍如何将一个扁平的数组对象，根据其level字段，转换为嵌套的树形结构。 挑战在于数组中对象的层级关系并非直接表达，而是通过level字段隐式表示。 我们需要根据level值判断对象的父子关系，最终构建清晰的嵌套结构。

假设我们有一个如下扁平数组：

const chapters = [{
  level: '1',
  content: '案例概况010000-170900',
  url: '案例概况010000-170900.html'
},{
  level: '2',
  content: '案例概况010100-170901',
  url: '案例概况010100-170901.html'
},{
  level: '2',
  content: '案例概况010200-170902',
  url: '案例概况010200-170902.html'
},{
  level: '3',
  content: '案例概况010201-170903',
  url: '案例概况010201-170903.html'
},{
  level: '1',
  content: '案例概况020000-170904',
  url: '案例概况020000-170904.html'
},{
  level: '1',
  content: '案例概况030000-170905',
  url: '案例概况030000-170905.html'
},{
  level: '2',
  content: '案例概况030100-170906',
  url: '案例概况030100-170906.html'
}];

我们的目标是将其转换为如下嵌套结构：

文心快码

文心快码（Comate）是百度推出的一款AI辅助编程工具

下载

const tree = [{
  level: '1',
  content: '案例概况010000-170900',
  url: '案例概况010000-170900.html',
  children: [{
    level: '2',
    content: '案例概况010100-170901',
    url: '案例概况010100-170901.html',
    children: []
  },{
    level: '2',
    content: '案例概况010200-170902',
    url: '案例概况010200-170902.html',
    children: [{
      level: '3',
      content: '案例概况010201-170903',
      url: '案例概况010201-170903.html',
      children: []
    }]
  }]
},{
  level: '1',
  content: '案例概况020000-170904',
  url: '案例概况020000-170904.html',
  children: []
},{
  level: '1',
  content: '案例概况030000-170905',
  url: '案例概况030000-170905.html',
  children: [{
    level: '2',
    content: '案例概况030100-170906',
    url: '案例概况030100-170906.html',
    children: []
  }]
}];

一种高效的解决方案是使用迭代和辅助数组parents跟踪当前处理的父节点。 以下代码展示了实现：

function createTreeByList(arr) {
  const tree = [];
  const parents = [];

  arr.forEach(item => {
    item.children = []; // 初始化children属性

    while (parents.length > 0 && +parents[parents.length - 1].level >= +item.level) {
      parents.pop();
    }

    if (parents.length === 0) {
      tree.push(item);
    } else {
      parents[parents.length - 1].children.push(item);
    }
    parents.push(item);
  });

  return tree;
}

const tree = createTreeByList(chapters);
console.log(JSON.stringify(tree, null, 2));

这段代码利用while循环找到合适的父节点，并将其添加到children属性中。 parents数组确保了父子关系的正确维护。 通过这种迭代方法，我们高效地将扁平数组转换为嵌套树形结构，避免了递归的开销。