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函数来检测t1t2是否存储了相同的值。
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"
    "fmt"
)

func dfs(t *tree.Tree, ch chan int) {
    if t == nil {
        return
    }	
    dfs(t.Left, ch)
    ch <- t.Value
    dfs(t.Right, ch)
}

// Walk 步进 tree t 将所有的值从 tree 发送到 channel ch
func Walk(t *tree.Tree, ch chan int) {
    dfs(t, ch)
    close(ch)
}

// Same 检测树 t1 和 t2 是否含有相同的值。
func Same(t1, t2 *tree.Tree) bool {
    ch1, ch2 := make(chan int), make(chan int)
    mp1, mp2 := make(map[int]bool), make(map[int]bool)
    go Walk(t1, ch1)
    go Walk(t2, ch2)
    for v1 := range ch1 {
        mp1[v1] = true
    }
    for v2 := range ch2 {
        mp2[v2] = true
    }
    for i := range mp1 {
        if mp2[i] != true {
            return false
        }
    }
    return true
}

func testWalk() {
    ch := make(chan int)
    go Walk(buildTree(), ch)
    for x := range ch {
        fmt.Println(x)
    } 
}

func main() {
    //testWalk()
    fmt.Println(Same(tree.New(1), tree.New(2)))
    fmt.Println(Same(tree.New(1), tree.New(1)))
}
posted @ 2024-02-18 16:19  Nephna  阅读(90)  评论(0编辑  收藏  举报