【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)
相关推荐
编程小猹16 分钟前
学习golang语言时遇到的难点语法
学习·golang·xcode
沈霁晨43 分钟前
Ruby语言的Web开发
开发语言·后端·golang
代码驿站5204 小时前
JavaScript语言的软件工程
开发语言·后端·golang
行路见知5 小时前
3.1 Go函数调用过程
golang
一只会飞的猪_5 小时前
国密加密golang加密,java解密
java·开发语言·golang
C++小厨神7 小时前
MATLAB语言的编程范式
开发语言·后端·golang
小柴狗7 小时前
MAC 地址转换为标准大写格式
golang
冯萦岚8 小时前
R语言的图形用户界面
开发语言·后端·golang
{⌐■_■}8 小时前
【GORM】事务,嵌套事务,保存点事务的使用,简单电商平台go案例
开发语言·jvm·后端·mysql·golang
C++小厨神8 小时前
Kotlin语言的正则表达式
开发语言·后端·golang