【Golang】Go语言编程思想(六):Channel,第三节,使用Channel实现树的遍历

使用 Channel 实现树的遍历

tree

在此处简单回顾一下之前学过的二叉树遍历,首先新建一个名为 tree 的目录,并在其下对文件和子目录进行如下组织:

其中 node.go 存放的是 Node 的定义:

go 复制代码
package tree

import "fmt"

type Node struct {
	Value       int
	Left, Right *Node
}

func (node Node) Print() {
	fmt.Print(node.Value, " ")
}

func (node *Node) SetValue(value int) {
	if node == nil {
		fmt.Println("Setting Value to nil" +
			" node. Ignored.")
		return
	}
	node.Value = value
}

func CreateNode(value int) *Node { return &Node{Value: value} }

// 👆 构建 Node 的工厂函数, Golang 支持返回局部变量的地址

traverse.go 存放的是树的遍历方法,使用的方法是中序遍历,下例还将展示如何以函数作为中序遍历的参数在 Golang 当中实现在树的遍历过程中的某些结点行为(本例实现行为是,当遍历到某个结点时,将其打印出来):

go 复制代码
package tree

import "fmt"

func (node *Node) Traverse() {
	node.TraverseFunc(func(n *Node) {
		n.Print()
	})
	fmt.Println()
}

func (node *Node) TraverseFunc(f func(*Node)) {
	if node == nil {
		return
	}
	node.Left.TraverseFunc(f)
	f(node)
	node.Right.TraverseFunc(f)
}

最后,entry.go 建立了一棵树:

go 复制代码
package main

import (
	"fmt"
	"learngo/tree"
)

func main() {
	var root tree.Node
	root = tree.Node{Value: 3}
	root.Left = &tree.Node{}
	root.Right = &tree.Node{5, nil, nil}
	root.Right.Left = new(tree.Node)
	root.Left.Right = tree.CreateNode(2)
	root.Right.Left.SetValue(4)

	fmt.Print("In-order traversal: ")
	root.Traverse()
}

使用 Channel 统计最大的结点下标

此处我们实现一个最简单的 Channel 遍历树的应用,即统计最大的结点下标,在 traverse.go 当中编写如下函数:

go 复制代码
// 👇 使用 Channel 来完成树的遍历
func (node *Node) TraverseWithChannel() chan *Node {
	// 调用 TraverseWithChannel 获得的返回值是 chan *Node
	out := make(chan *Node)
	go func() {
		node.TraverseFunc(func(node *Node) {
			out <- node
		})
		close(out)
	}()
	return out
}

函数的返回值是 chan *Node ,即一个类行为 *Node 的通道。函数体当中的行为如下:

首先使用 make 方法新建一个通道,之后开启协程,将结点的值写入到通道当中。完成写入之后使用 close 方法关闭通道。

协程启动之后将通道返回。

在 main 函数中添加以下片段进行验证:

go 复制代码
c := root.TraverseWithChannel()
maxNode := 0 // 统计最大的结点数
for node := range c {			// 不断地接受来自 channel 的信息
	if node.Value > maxNode {
		maxNode = node.Value
	}
}
fmt.Println("Max Node Value:", maxNode)
相关推荐
大卡尔15 分钟前
Reviewbot 开源 | 这些写 Go 代码的小技巧,你都知道吗?
golang·go·静态检查·reviewbot
闲人怪喵4 小时前
/usr/local/go/bin/go: cannot execute binary file: Exec format error
开发语言·后端·golang
免檒11 小时前
GO并发编程
开发语言·golang
MavenTalk14 小时前
新手上路,学Go还是Python
开发语言·python·微服务·golang
YGGP15 小时前
【Golang】Go语言编程思想(三):资源管理和出错处理
golang
007php00717 小时前
php项目的sdk封装成composer包的创建与发版
运维·开发语言·nginx·golang·github·php·composer
007php00718 小时前
go语言zero框架对接阿里云消息队列MQ的rabbit的配置与调用
开发语言·后端·python·gpt·阿里云·golang·github
Ai 编码助手19 小时前
Go 协程上下文切换的代价
服务器·golang
Clown9519 小时前
go-zero(十二)消息队列
golang·go zero