package main
import "code.google.com/p/go-tour/tree"
import "fmt"
// Walk 步进 tree t 将所有的值从 tree 发送到 channel ch。
func Walk(t *tree.Tree, ch chan int) {
transValue(t, ch)
close(ch)
}
func transValue(t *tree.Tree, ch chan int) {
if t != nil {
transValue(t.Left, ch)
ch <- t.Value
transValue(t.Right, ch)
}
}
// Same 检测树 t1 和 t2 是否含有相同的值。
func Same(t1, t2 *tree.Tree) bool {
ch1 := make(chan int)
ch2 := make(chan int)
go Walk(t1, ch1)
go Walk(t2, ch2)
for i := range ch1 {
if i != <-ch2 {
return false
}
}
return true
}
func main() {
ch := make(chan int)
go Walk(tree.New(1), ch)
for v := range ch {
fmt.Println(v)
}
// 比较两个tree的value值是否相等
fmt.Println(Same(tree.New(1), tree.New(1)))
fmt.Println(Same(tree.New(1), tree.New(2)))
}
练习:等价二叉树
1. 实现 Walk
函数。
2. 测试 Walk
函数。
函数 tree.New(k)
构造了一个随机结构的二叉树,保存了值 `k`,`2k`,`3k`,...,`10k`。 创建一个新的 channel ch
并且对其进行步进:
go Walk(tree.New(1), ch)
然后从 channel 中读取并且打印 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。