学习 Go 语言 0x0c：《Go 语言之旅》中练习等价二叉查找树

本文链接：https://blog.csdn.net/u013553529/article/details/88920480

★ 题目

https://tour.go-zh.org/concurrency/7
https://tour.go-zh.org/concurrency/8

练习: 等价二叉查找树
不同二叉树的叶节点上可以保存相同的值序列。例如，以下两个二叉树都保存了序列 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"
import "fmt"

// Walk 步进 tree t 将所有的值从 tree 发送到 channel ch。
func Walk(t *tree.Tree, ch chan int) {
	if t.Left != nil {
		Walk(t.Left, ch)
	}
	ch <- t.Value
	if t.Right != nil {
		Walk(t.Right, ch)
	}	
}

func TestWalk() {
	ch := make(chan int, 10)
	k := 2
	go Walk(tree.New(k), ch)
	for i := 1; i <= 10; i++ {
		val := <-ch
		if val != i * k {
			fmt.Println("FAIL: Walk() goes wrong.")
			return
		}
	}
	fmt.Println("PASS: Walk() goes well")
}

// Same 检测树 t1 和 t2 是否含有相同的值。
func Same(t1, t2 *tree.Tree) bool {
	ch1 := make(chan int, 10)
	ch2 := make(chan int, 10)
	go Walk(t1, ch1)
	go Walk(t2, ch2)
	for i := 0; i < 10; i++ {
		n1, n2 := <-ch1, <-ch2
		if n1 != n2 {
			return false
		}
	}
	return true
}

func main() {
	ch := make(chan int, 10)
	k := 1
	go Walk(tree.New(k), ch)
	for i := 0; i < 10; i++ {
		fmt.Println(<-ch)
	}
	TestWalk()
	
	same := Same(tree.New(1), tree.New(1))
	if same {
		fmt.Println("PASS : they are same")
	} else {
		fmt.Println("FAIL : they should be same.")
	}
	
	same = Same(tree.New(1), tree.New(2))
	if !same {
		fmt.Println("PASS : they are not same")
	} else {
		fmt.Println("FAIL : they should be not same")
	}
}

输出：

1
2
3
4
5
6
7
8
9
10
PASS: Walk() goes well
PASS : they are same
PASS : they are not same

★ 用 `testing`包测试 `Walk` 函数

注1：代码是在本地环境中执行的，不是在线执行的。

注2：测试代码和被测试代码必须在同一个目录（包）下。

注3：下面的例子是将两个 go 文件放到 mypkg 包下面了，也可以放到 main 包下面，即，把 package mypkg 改为 package main。注意：一个包内不能有2个main函数。

文件：src/mypkg/binary_tree01.go

package mypkg

import "golang.org/x/tour/tree"
import "fmt"

// Walk 步进 tree t 将所有的值从 tree 发送到 channel ch。
func Walk(t *tree.Tree, ch chan int) {
    if t.Left != nil {
        Walk(t.Left, ch)
    }
    ch <- t.Value
    if t.Right != nil {
        Walk(t.Right, ch)
    }
}

func main() {
    ch := make(chan int, 10)
    k := 1
    go Walk(tree.New(k), ch)
    for i := 0; i < 10; i++ {
        fmt.Println(<-ch)
    }
}

文件：src/mypkg/binary_tree01_test.go

package mypkg

import "golang.org/x/tour/tree"
import "testing"

func TestWalk(t *testing.T) {
    ch := make(chan int, 10)
    k := 2
    go Walk(tree.New(k), ch)
    for i := 1; i <= 10; i++ {
        val := <-ch
        if val != i * k {
            t.Errorf("Expect %d, but get wrong %d\n", i * k, val)
        }
    }
}