这个是A Tour of Go这个教程里面的第十个练习,原题如下:
不同二叉树的叶节点上可以保存相同的值序列。例如,以下两个二叉树都保存了序列 `1,1,2,3,5,8,13`。
在大多数语言中,检查两个二叉树是否保存了相同序列的函数都相当复杂。 我们将使用 Go 的并发和信道来编写一个简单的解法。
本例使用了
tree
包,它定义了类型:type Tree struct { Left *Tree Value int Right *Tree }
1. 实现
Walk
函数。2. 测试
Walk
函数。函数
tree.New(k)
用于构造一个随机结构的已排序二叉查找树,它保存了值k
,2k
,3k
, …,10k
。创建一个新的信道
ch
并且对其进行步进:go Walk(tree.New(1), ch)
然后从信道中读取并打印 10 个值。应当是数字
1, 2, 3, ..., 10
。3. 用
Walk
实现Same
函数来检测t1
和t2
是否存储了相同的值。4. 测试
Same
函数。
Same(tree.New(1), tree.New(1))
应当返回true
,而Same(tree.New(1), tree.New(2))
应当返回false
。
Tree
的文档可在这里找到。
原题代码模板如下:
package main
import "golang.org/x/tour/tree"
// Walk 步进 tree t 将所有的值从 tree 发送到 channel ch。
func Walk(t *tree.Tree, ch chan int)
// Same 检测树 t1 和 t2 是否含有相同的值。
func Same(t1, t2 *tree.Tree) bool
func main() {
}
这个Same
实现的思想是创建两个信道,通过Walk
函数分别遍历两棵树,并把结果分别传入两个信道,然后比较两个信道的结果是否相同。实现如下:
package main
import (
"fmt"
"golang.org/x/tour/tree"
)
// Walk 步进 tree t 将所有的值从 tree 发送到 channel ch。
func Walk(t *tree.Tree, ch chan int){
if t == nil {
return
}
Walk(t.Left, ch)
ch <- t.Value
Walk(t.Right, ch)
}
// Same 检测树 t1 和 t2 是否含有相同的值。
func Same(t1, t2 *tree.Tree) bool{
ch1, ch2 := make(chan int), make(chan int)
go Walk(t1, ch1)
go Walk(t2, ch2)
for i := 0; i < 10; i++{
x, y := <-ch1, <-ch2
if x != y {
return false
}
}
return true
}
func main() {
ch := make(chan int)
//从信道中打印10个值
go Walk(tree.New(1), ch)
for i := 0; i < 10; i++ {
fmt.Println(<-ch)
}
fmt.Println(Same(tree.New(1), tree.New(1)))
fmt.Println(Same(tree.New(1), tree.New(2)))
}
输出如下:
1
2
3
4
5
6
7
8
9
10
true
false