LingKangSg的go学习日记9之DAG

当使用Go语言实现DAG（有向无环图）数据结构时，可以使用邻接表的方式来表示图的结构。下面是一个简单的DAG数据结构的Go实现示例：

```go
package main

import (
	"fmt"
)

type Node struct {
	ID       int
	Children []*Node
}

type DAG struct {
	Nodes []*Node
}

func (d *DAG) AddNode(id int) {
	node := &Node{
		ID: id,
	}
	d.Nodes = append(d.Nodes, node)
}

func (d *DAG) AddEdge(parentID, childID int) error {
	parentNode, childNode := d.getNodeByID(parentID), d.getNodeByID(childID)
	if parentNode == nil || childNode == nil {
		return fmt.Errorf("Node not found")
	}
	parentNode.Children = append(parentNode.Children, childNode)
	return nil
}

func (d *DAG) getNodeByID(id int) *Node {
	for _, node := range d.Nodes {
		if node.ID == id {
			return node
		}
	}
	return nil
}

func main() {
	dag := &DAG{}

	// 添加节点
	dag.AddNode(1)
	dag.AddNode(2)
	dag.AddNode(3)
	dag.AddNode(4)
	dag.AddNode(5)

	// 添加边
	dag.AddEdge(1, 2)
	dag.AddEdge(1, 3)
	dag.AddEdge(2, 4)
	dag.AddEdge(3, 4)
	dag.AddEdge(4, 5)

	// 打印节点和边
	for _, node := range dag.Nodes {
		fmt.Printf("Node %d has children: ", node.ID)
		for _, child := range node.Children {
			fmt.Printf("%d ", child.ID)
		}
		fmt.Println()
	}
}
```

在上面的示例中，我们定义了`Node`结构体来表示图中的节点，每个节点包含一个ID和一个指向其子节点的切片。`DAG`结构体表示整个有向无环图，其中包含一个节点切片。我们使用`AddNode`方法来添加节点，使用`AddEdge`方法来添加边。`getNodeByID`方法用于根据节点ID获取节点对象。

在`main`函数中，我们创建了一个DAG对象，并添加了一些节点和边。然后，我们遍历节点并打印每个节点的子节点。

这只是一个简单的DAG数据结构的实现示例，你可以根据自己的需求进行扩展和修改。