使用Go语言将扁平化表格数据高效转换为树形结构

type Node struct {
    name     string
    children []*Node
}

var nodeTable = map[string]*Node{}

var root *Node

func add(id, name, parentId string) {
    node := &Node{name: name, children: []*Node{}} // 创建新节点

    if parentId == "0" { // 如果父ID为"0"，则此节点是根节点
        root = node
    } else {
        // 查找父节点
        parent, ok := nodeTable[parentId]
        if !ok {
            fmt.Printf("错误：未找到父节点ID %v\n", parentId)
            return
        }
        // 将新节点添加到父节点的子节点列表中
        parent.children = append(parent.children, node)
    }

    // 将新节点添加到全局节点表中，以便后续查找
    nodeTable[id] = node
}

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

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

func scan() {
    reader := bufio.NewReader(os.Stdin) // 从标准输入读取
    lineCount := 0
    for {
        lineCount++
        line, err := reader.ReadString('\n') // 读取一行
        if err == io.EOF { // 读取到文件末尾
            break
        }
        if err != nil {
            fmt.Printf("读取输入时发生错误: %v\n", err)
            return
        }

        tokens := strings.Fields(line) // 按空格分割字符串
        if len(tokens) != 3 { // 检查字段数量
            fmt.Printf("第 %v 行输入格式错误，跳过: %v\n", lineCount, strings.TrimSpace(line))
            continue
        }
        // 调用add函数构建树
        add(tokens[0], tokens[1], tokens[2])
    }
}

func showNode(node *Node, prefix string) {
    if node == nil {
        return
    }
    // 打印当前节点，并根据层级添加缩进
    fmt.Printf("%s%s\n", prefix, node.name)

    // 递归打印所有子节点
    for _, n := range node.children {
        showNode(n, prefix+"--") // 子节点的缩进增加
    }
}

func show() {
    if root == nil {
        fmt.Printf("错误：根节点未找到，无法展示树结构。\n")
        return
    }
    fmt.Printf("树形结构结果:\n")
    showNode(root, "") // 从根节点开始展示，初始缩进为空
}

package main

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

// Node 结构体定义树中的一个节点
type Node struct {
    name     string
    children []*Node
}

// nodeTable 用于通过ID快速查找节点
var nodeTable = map[string]*Node{}

// root 存储树的根节点
var root *Node

// add 函数用于创建新节点并将其添加到树中
func add(id, name, parentId string) {
    node := &Node{name: name, children: []*Node{}}

    if parentId == "0" {
        root = node
    } else {
        parent, ok := nodeTable[parentId]
        if !ok {
            fmt.Printf("错误：未找到父节点ID %v，无法添加节点 %v (%v)\n", parentId, name, id)
            return
        }
        parent.children = append(parent.children, node)
    }
    nodeTable[id] = node
}

// scan 函数从标准输入读取数据并构建树
func scan() {
    reader := bufio.NewReader(os.Stdin)
    lineCount := 0
    for {
        lineCount++
        line, err := reader.ReadString('\n')
        if err == io.EOF {
            break
        }
        if err != nil {
            fmt.Printf("读取输入时发生错误: %v\n", err)
            return
        }
        tokens := strings.Fields(line)
        if len(tokens) != 3 {
            fmt.Printf("第 %v 行输入格式错误，期望3个字段，实际 %d 个: %v\n", lineCount, len(tokens), strings.TrimSpace(line))
            continue
        }
        add(tokens[0], tokens[1], tokens[2])
    }
}

// showNode 递归地展示树的结构，带有缩进
func showNode(node *Node, prefix string) {
    if node == nil {
        return
    }
    fmt.Printf("%s%s\n", prefix, node.name)
    for _, n := range node.children {
        showNode(n, prefix+"--")
    }
}

// show 函数作为展示树的入口
func show() {
    if root == nil {
        fmt.Printf("错误：根节点未找到，无法展示树结构。\n")
        return
    }
    fmt.Printf("树形结构结果:\n")
    showNode(root, "")
}

func main() {
    fmt.Println("请输入表格数据 (OrgID OrgName parentID)，每行一个，输入完成后按 Ctrl+D 结束:")
    scan()
    fmt.Println("\n数据读取完毕。")
    show()
    fmt.Println("\n程序执行结束。")
}