使用Go语言将扁平化数据构建为层级树结构

本教程详细阐述了如何利用go语言，将具有父子关系的扁平化表格数据高效地转换为内存中的树形结构。通过定义简洁的节点类型、运用哈希映射实现父节点的快速查找与关联，以及设计递归函数进行树的构建与可视化展示，我们提供了一套完整的解决方案，旨在帮助开发者理解并实现此类数据转换，并附带了可运行的代码示例。

1. 理解树形数据结构与扁平化数据

在许多应用场景中，数据天然具有层级关系，例如组织架构、文件系统或评论回复链。然而，这些层级数据在数据库或CSV文件中常常以扁平化的表格形式存储，每行记录包含一个唯一的ID、名称以及一个指向其父节点的ID。我们的目标就是将这种扁平化的父子关系数据转换为内存中的树形结构，以便于进行层级遍历、查找和展示。

例如，以下表格展示了一个简化的部门层级结构：

其中 parentID 为 "0" 的节点被视为根节点。我们希望将其转换为如下所示的树形结构：

Dept
--subDept1
----sub_subDept
------gran_subDept
--subDept2

2. 数据模型定义

为了在Go语言中表示树形结构，我们需要定义一个节点类型，该节点包含其自身的名称以及一个指向其所有子节点的切片。同时，为了高效地根据ID查找节点，我们还需要一个全局的哈希映射。

package main

import (
    "bufio"
    "fmt"
    "io"
    "os"
    "strings"
)

// Node 定义了树中的一个节点
type Node struct {
    Name     string    // 节点的名称
    Children []*Node   // 节点的子节点列表
}

var (
    // nodeTable 用于通过ID快速查找已创建的节点
    nodeTable = map[string]*Node{}
    // root 指向树的根节点。本示例假设只有一个根节点。
    root      *Node
)

解释：

• Node 结构体：Name 存储节点的显示名称，Children 是一个 *Node 类型的切片，用于存储当前节点的所有子节点。
• nodeTable：这是一个 map[string]*Node，其键是节点的唯一ID（如OrgID），值是指向该节点的指针。这个映射在构建树时至关重要，它允许我们通过 parentID 快速找到对应的父节点。
• root：一个 *Node 类型的全局变量，用于存储整个树的根节点。本示例代码假定只有一个根节点，其 parentID 为 "0"。

3. 核心逻辑实现

3.1 添加节点 (add 函数)

add 函数负责创建新节点，并将其正确地插入到树中。它根据 parentID 判断节点是根节点还是普通子节点。

// add 函数用于创建新节点并将其添加到树结构中
// id: 当前节点的唯一标识
// name: 当前节点的显示名称
// parentId: 当前节点的父节点标识
func add(id, name, parentId string) {
    fmt.Printf("add: id=%v name=%v parentId=%v\n", id, name, parentId)

    // 创建一个新的节点实例
    node := &Node{Name: name, Children: []*Node{}}

    // 根据 parentId 判断节点类型
    if parentId == "0" {
        // 如果 parentId 为 "0"，则当前节点是根节点
        root = node
    } else {
        // 否则，当前节点是子节点，需要找到其父节点
        parent, ok := nodeTable[parentId]
        if !ok {
            // 如果父节点不存在，打印错误并跳过
            fmt.Printf("add: parentId=%v: not found, skipping node %v\n", parentId, id)
            return
        }
        // 将当前节点添加到父节点的 Children 列表中
        parent.Children = append(parent.Children, node)
    }

    // 将新创建的节点添加到 nodeTable 中，以便后续查找
    nodeTable[id] = node
}

解释：

1. 创建一个新的 Node 实例，并初始化其 Name 和空的 Children 切片。
2. 如果 parentId 是 "0"，则将此节点设置为全局 root。
3. 如果 parentId 不是 "0"，则尝试从 nodeTable 中查找对应的父节点。
   • 如果父节点不存在，表示数据可能存在问题（例如，父节点ID引用了不存在的节点），此时会打印错误并跳过当前节点的处理。
   • 如果找到父节点，则将新创建的节点添加到父节点的 Children 切片中。
4. 最后，无论节点是根还是子，都将其存储到 nodeTable 中，以便其他节点可以将其作为父节点引用。

3.2 数据输入与解析 (scan 函数)

scan 函数负责从标准输入读取数据，解析每一行，并调用 add 函数来构建树。

// scan 函数从标准输入读取数据，解析每行并调用 add 函数构建树
func scan() {
    input := os.Stdin
    reader := bufio.NewReader(input)
    lineCount := 0
    for {
        lineCount++
        line, err := reader.ReadString('\n') // 读取一行直到换行符
        if err == io.EOF {
            break // 文件结束
        }
        if err != nil {
            fmt.Printf("error reading lines: %v\n", err)
            return
        }
        // 使用 strings.Fields 分割字符串，默认按空格或制表符分割
        tokens := strings.Fields(line)
        if t := len(tokens); t != 3 {
            // 检查每行是否有3个字段 (OrgID, OrgName, parentID)
            fmt.Printf("bad input line %v: tokens=%d [%v], expected 3 fields\n", lineCount, t, line)
            continue
        }
        // 调用 add 函数处理当前行数据
        add(tokens[0], tokens[1], tokens[2])
    }
}

解释：

• 使用 bufio.NewReader 从 os.Stdin 读取输入。
• 循环读取每一行，直到遇到 io.EOF (文件结束)。
• strings.Fields(line) 会将一行字符串按空白字符（空格、制表符等）分割成字符串切片。
• 检查 tokens 的长度是否为3，确保输入格式正确。
• 将解析出的 OrgID、OrgName 和 parentID 传递给 add 函数。

3.3 树形结构展示 (showNode 和 show 函数)

showNode 函数使用递归的方式遍历树并打印节点名称，通过 prefix 参数实现层级缩进。show 函数作为入口，调用 showNode 从根节点开始展示。

// showNode 递归地显示树中的每个节点及其子节点
// node: 当前要显示的节点
// prefix: 用于缩进的字符串，表示当前节点的层级
func showNode(node *Node, prefix string) {
    if prefix == "" {
        // 根节点不加前缀，单独打印
        fmt.Printf("%v\n", node.Name)
    } else {
        // 子节点加上缩进前缀
        fmt.Printf("%v%v\n", prefix, node.Name)
    }
    // 递归遍历所有子节点
    for _, n := range node.Children {
        showNode(n, prefix+"--") // 子节点的缩进增加 "--"
    }
}

// show 函数是显示树结构的入口
func show() {
    if root == nil {
        fmt.Printf("show: root node not found, tree is empty\n")
        return
    }
    fmt.Printf("\nRESULT:\n")
    showNode(root, "") // 从根节点开始显示，初始前缀为空
}

解释：

• showNode 是一个递归函数。它首先打印当前节点的名称，并根据 prefix 参数决定是否添加缩进。
• 然后，它遍历当前节点的所有 Children，并对每个子节点递归调用 showNode，同时将 prefix 增加 "--"，以实现逐级缩进的效果。
• show 函数简单地检查 root 是否存在，如果存在，则调用 showNode 开始展示。

4. 完整代码示例

将上述所有部分组合起来，构成一个完整的Go程序：

package main

import (
    "bufio"
    "fmt"
    "io"
    "os"
    "strings"
)

// Node 定义了树中的一个节点
type Node struct {
    Name     string    // 节点的名称
    Children []*Node   // 节点的子节点列表
}

var (
    // nodeTable 用于通过ID快速查找已创建的节点
    nodeTable = map[string]*Node{}
    // root 指向树的根节点。本示例假设只有一个根节点。
    root      *Node
)

// add 函数用于创建新节点并将其添加到树结构中
// id: 当前节点的唯一标识
// name: 当前节点的显示名称
// parentId: 当前节点的父节点标识
func add(id, name, parentId string) {
    fmt.Printf("add: id=%v name=%v parentId=%v\n", id, name, parentId)

    // 创建一个新的节点实例
    node := &Node{Name: name, Children: []*Node{}}

    // 根据 parentId 判断节点类型
    if parentId == "0" {
        // 如果 parentId 为 "0"，则当前节点是根节点
        root = node
    } else {
        // 否则，当前节点是子节点，需要找到其父节点
        parent, ok := nodeTable[parentId]
        if !ok {
            // 如果父节点不存在，打印错误并跳过
            fmt.Printf("add: parentId=%v: not found, skipping node %v\n", parentId, id)
            return
        }
        // 将当前节点添加到父节点的 Children 列表中
        parent.Children = append(parent.Children, node)
    }

    // 将新创建的节点添加到 nodeTable 中，以便后续查找
    nodeTable[id] = node
}

// scan 函数从标准输入读取数据，解析每行并调用 add 函数构建树
func scan() {
    input := os.Stdin
    reader := bufio.NewReader(input)
    lineCount := 0
    for {
        lineCount++
        line, err := reader.ReadString('\n') // 读取一行直到换行符
        if err == io.EOF {
            break // 文件结束
        }
        if err != nil {
            fmt.Printf("error reading lines: %v\n", err)
            return
        }
        // 使用 strings.Fields 分割字符串，默认按空格或制表符分割
        tokens := strings.Fields(line)
        if t := len(tokens); t != 3 {
            // 检查每行是否有3个字段 (OrgID, OrgName, parentID)
            fmt.Printf("bad input line %v: tokens=%d [%v], expected 3 fields\n", lineCount, t, line)
            continue
        }
        // 调用 add 函数处理当前行数据
        add(tokens[0], tokens[1], tokens[2])
    }
}

// showNode 递归地显示树中的每个节点及其子节点
// node: 当前要显示的节点
// prefix: 用于缩进的字符串，表示当前节点的层级
func showNode(node *Node, prefix string) {
    if prefix == "" {
        // 根节点不加前缀，单独打印
        fmt.Printf("%v\n", node.Name)
    } else {
        // 子节点加上缩进前缀
        fmt.Printf("%v%v\n", prefix, node.Name)
    }
    // 递归遍历所有子节点
    for _, n := range node.Children {
        showNode(n, prefix+"--") // 子节点的缩进增加 "--"
    }
}

// show 函数是显示树结构的入口
func show() {
    if root == nil {
        fmt.Printf("show: root node not found, tree is empty\n")
        return
    }
    fmt.Printf("\nRESULT:\n")
    showNode(root, "") // 从根节点开始显示，初始前缀为空
}

// main 函数是程序的入口点
func main() {
    fmt.Printf("main: reading input from stdin\n")
    scan() // 读取并解析输入数据
    fmt.Printf("main: reading input from stdin -- done\n")
    show() // 显示构建好的树结构
    fmt.Printf("main: end\n")
}

5. 如何运行与测试

1. 保存代码： 将上述完整代码保存为 tree_builder.go 文件。

2. 准备输入数据： 创建一个文本文件，例如 input.txt，将您的扁平化表格数据粘贴进去。确保每行包含三个由空格分隔的字段：OrgID OrgName parentID。

   A001 Dept 0
   A002 subDept1 A001
   A003 sub_subDept A002
   A006 gran_subDept A003
   A004 subDept2 A001

3. 编译程序： 打开终端或命令行，导航到 tree_builder.go 文件所在的目录，然后执行以下命令：

   go build -o tree_builder

   这会生成一个名为 tree_builder (Windows 上是 tree_builder.exe) 的可执行文件。

4. 运行程序： 通过管道将 input.txt 的内容作为标准输入传递给程序：

   cat input.txt | ./tree_builder

   或者在 Windows 上：

   type input.txt | .\tree_builder.exe

   您将看到程序打印的调试信息和最终的树形结构输出：

   main: reading input from stdin
   add: id=A001 name=Dept parentId=0
   add: id=A002 name=subDept1 parentId=A001
   add: id=A003 name=sub_subDept parentId=A002
   add: id=A006 name=gran_subDept parentId=A003
   add: id=A004 name=subDept2 parentId=A001
   main: reading input from stdin -- done
   
   RESULT:
   Dept
   --subDept1
   ----sub_subDept
   ------gran_subDept
   --subDept2
   main: end

6. 注意事项与潜在优化

• 单一根节点假设： 当前代码假定树只有一个根节点，即只有一个 parentID 为 "0" 的记录。如果存在多个根节点，root 变量将只保存最后处理的那个根节点。要支持多个根节点，可以将 root 修改为 []*Node 类型，并在 parentId == "0" 时将节点添加到这个切片中。
• 输入数据顺序： 尽管 nodeTable 的存在使得 add 函数能够处理父节点在其子节点之后出现的情况（只要父节点最终被处理），但在实际应用中，如果父节点总是先于其子节点出现，处理逻辑会更直观且可能略高效。
• 错误处理：
  • 循环引用： 当前代码没有检测循环引用（例如 A 的父是 B，

# node  # go  # windows  # go语言  # app  # csv  # ai  # win  # 递归函数  # csv文件  # 架构  # EOF  # String  # 全局变量  # 字符串  # 结构体  # 递归  # 循环  # 指针  # 数据结构 

相关文章： 制作网站的软件下载免费,今日头条开宝箱老是需要下载怎么回事？  动图在线制作网站有哪些,滑动动图图集怎么做？  如何在云主机快速搭建网站站点？  制作网站哪家好,cc、.co、.cm哪个域名更适合做网站？  常州企业建站如何选择最佳模板？  如何高效配置IIS服务器搭建网站？  胶州企业网站制作公司,青岛石头网络科技有限公司怎么样？  上海网站制作网页,上海本地的生活网站有哪些？最好包括生活的各个方面的？  平台云上自助建站如何快速打造专业网站？  c++ stringstream用法详解_c++字符串与数字转换利器  网站制作费用多少钱,一个网站的运营，需要哪些费用？  广州建站公司哪家好？十大优质服务商推荐  岳西云建站教程与模板下载_一站式快速建站系统操作指南  弹幕视频网站制作教程下载,弹幕视频网站是什么意思？  合肥做个网站多少钱,合肥本地有没有比较靠谱的交友平台？  ,网页ppt怎么弄成自己的ppt？  建站之星logo尺寸如何设置最合适？  表情包在线制作网站免费,表情包怎么弄？  免费网站制作appp,免费制作app哪个平台好？  IOS倒计时设置UIButton标题title的抖动问题  建站之星在线版空间：自助建站+智能模板一键生成方案  如何在建站宝盒中设置产品搜索功能？  如何通过建站之星自助学习解决操作问题？  如何安全更换建站之星模板并保留数据？  建站VPS选购需注意哪些关键参数？  陕西网站制作公司有哪些,陕西凌云电器有限公司官网？  为什么Go需要go mod文件_Go go mod文件作用说明  西安大型网站制作公司,西安招聘网站最好的是哪个？  SQL查询语句优化的实用方法总结  网站制作公司哪里好做,成都网站制作公司哪家做得比较好，更正规？  C#怎么使用委托和事件 C# delegate与event编程方法  网站微信制作软件,如何制作微信链接？  企业在线网站设计制作流程,想建设一个属于自己的企业网站，该如何去做？  c# F# 的 MailboxProcessor 和 C# 的 Actor 模型  山东网站制作公司有哪些,山东大源集团官网？  C#如何在一个XML文件中查找并替换文本内容  网站图片在线制作软件,怎么在图片上做链接？  网站制作哪家好,cc、.co、.cm哪个域名更适合做网站？  魔毅自助建站系统：模板定制与SEO优化一键生成指南  建站之星好吗？新手能否轻松上手建站？  建站主机助手选型指南：2025年热门推荐与高效部署技巧  如何快速搭建高效香港服务器网站？  浅析上传头像示例及其注意事项  定制建站方案优化指南：企业官网开发与建站费用解析  网页制作模板网站推荐,网页设计海报之类的素材哪里好？  如何自定义建站之星模板颜色并下载新样式？  建站主机与服务器功能差异如何区分？  头像制作网站在线制作软件,dw网页背景图像怎么设置？  如何在IIS管理器中快速创建并配置网站？  建站之星代理如何优化在线客服效率？