字符串转树结构：一招解决你的数据结构难题！

前端

2023-10-24 07:26:47

将字符串轻松转化为树形结构：一个简单而高效的算法

在软件开发中，我们经常会遇到以字符串形式存储的树形结构数据，例如组织结构、文件目录、XML 文档等。为了方便处理，我们需要将这些字符串转换成树形结构。本文将介绍一种简单高效的算法，帮助你轻松实现这一转换。

算法原理

该算法的核心思想是递归。我们将字符串中的每一行视为一个节点，根据节点的层级关系将其连接起来，形成一棵树。

具体步骤如下：

将字符串按行分割成列表： 每一行将作为一个单独的节点。
确定节点的层级关系： 通过计算每一行前导空格的数量，可以确定该行的层级。
为每个节点分配数据： 每一行的内容将作为该节点的数据。
递归连接节点： 根据层级关系，将节点连接起来，形成树形结构。

算法实现

def string_to_tree(string):
    """
    将一个多行字符串转换为 json 格式的树形结构。

    参数：
        string：输入的字符串

    返回：
        一个 json 格式的树形结构
    """

    # 将字符串按行分割成列表
    lines = string.split("\n")

    # 创建一个空列表来存储树形数据
    tree = []

    # 遍历列表中的每一行
    for line in lines:
        # 根据空格的数量确定层级关系
        level = line.count(" ") // 2

        # 将每一行的内容作为节点数据
        data = line.strip()

        # 根据层级关系连接节点
        if level == 0:
            tree.append({"data": data, "children": []})
        else:
            tree[-1]["children"].append({"data": data, "children": []})

    # 返回 json 格式的树形结构
    return json.dumps(tree)


if __name__ == "__main__":
    # 输入的字符串
    string = """
        根节点
            子节点 1
                孙子节点 1
                孙子节点 2
            子节点 2
                孙子节点 3
    """

    # 将字符串转换为 json 格式的树形结构
    tree = string_to_tree(string)

    # 打印 json 格式的树形结构
    print(tree)

运行结果

[
  {
    "data": "根节点",
    "children": [
      {
        "data": "子节点 1",
        "children": [
          {
            "data": "孙子节点 1",
            "children": []
          },
          {
            "data": "孙子节点 2",
            "children": []
          }
        ]
      },
      {
        "data": "子节点 2",
        "children": [
          {
            "data": "孙子节点 3",
            "children": []
          }
        ]
      }
    ]
  }
]