构建高效层级数据：Golang 树形结构选择与实现

type Node struct {
    ID       string
    Parent   *Node
    Children []*Node
    Data     interface{} // 可存储节点相关的数据
}

type Tree struct {
    Root *Node
    NodeMap map[string]*Node // 可选：用于快速查找节点
}

// 查找父节点
func (n *Node) GetParent() *Node {
    return n.Parent
}

// 查找子节点
func (n *Node) GetChildren() []*Node {
    return n.Children
}

// 通过ID查找节点 (需要 NodeMap)
func (t *Tree) FindNode(id string) *Node {
    if t.NodeMap == nil {
        return nil // 或者遍历整个树查找
    }
    node, ok := t.NodeMap[id]
    if !ok {
        return nil
    }
    return node
}

// 添加子节点
func (n *Node) AddChild(child *Node) {
    child.Parent = n
    n.Children = append(n.Children, child)
}

// 构建NodeMap
func (t *Tree) BuildNodeMap(node *Node) {
    if t.NodeMap == nil {
        t.NodeMap = make(map[string]*Node)
    }
    t.NodeMap[node.ID] = node
    for _, child := range node.Children {
        t.BuildNodeMap(child)
    }
}

package main

import "fmt"

func main() {
    // 创建根节点
    root := &Node{ID: "root", Data: "Root Node"}

    // 创建子节点
    child1 := &Node{ID: "child1", Data: "Child 1"}
    child2 := &Node{ID: "child2", Data: "Child 2"}

    // 添加子节点到根节点
    root.AddChild(child1)
    root.AddChild(child2)

    // 创建树
    tree := &Tree{Root: root}
    tree.BuildNodeMap(root)

    // 查找子节点
    children := root.GetChildren()
    fmt.Println("Children of root:", children)

    // 查找child1的父节点
    parent := child1.GetParent()
    fmt.Println("Parent of child1:", parent.ID)

    // 通过ID查找节点
    node := tree.FindNode("child2")
    fmt.Println("Node of child2:", node.ID)
}