Go语言并发二叉树遍历：通道关闭与等价性判断的优雅方案

package main

import (
    "fmt"
    "golang.org/x/tour/tree" // 假设这个包提供了tree.Tree结构和New函数
)

// Walk 函数将二叉树 t 的所有值发送到通道 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)
    }
    // 错误示范：如果在这里 close(ch)，会过早关闭通道
    // close(ch)
}

package main

import (
    "fmt"
    "golang.org/x/tour/tree" // 假设这个包提供了tree.Tree结构和New函数
)

// Walk 函数将二叉树 t 的所有值发送到通道 ch
// 并在所有值发送完毕后关闭通道。
func Walk(t *tree.Tree, ch chan int) {
    // defer close(ch) 确保通道在 Walk 函数返回时关闭。
    // 由于递归逻辑被封装在内部闭包中，这个 defer 会在所有递归完成后才执行。
    defer close(ch)

    // 定义一个内部闭包，用于执行实际的递归遍历
    var walk func(t *tree.Tree)
    walk = func(t *tree.Tree) {
        if t == nil {
            return
        }
        walk(t.Left)    // 遍历左子树
        ch <- t.Value   // 发送当前节点值
        walk(t.Right)   // 遍历右子树
    }
    walk(t) // 启动内部闭包的遍历
}

// Same 函数判断两棵二叉树 t1 和 t2 是否包含相同的值。
func Same(t1, t2 *tree.Tree) bool {
    ch1 := make(chan int)
    ch2 := make(chan int)

    // 启动两个 goroutine 并发遍历两棵树
    go Walk(t1, ch1)
    go Walk(t2, ch2)

    // 从两个通道中读取值并进行比较
    for {
        v1, ok1 := <-ch1 // 从 ch1 读取值
        v2, ok2 := <-ch2 // 从 ch2 读取值

        switch {
        case !ok1 && !ok2: // 两个通道都已关闭，且之前所有值都匹配
            return true
        case !ok1 || !ok2: // 一个通道关闭，另一个仍有值，表示不相等
            return false
        case v1 != v2:     // 值不匹配，表示不相等
            return false
        }
        // 如果两个通道都有值且值匹配，则继续循环
    }
}

package main

import (
    "fmt"
    "golang.org/x/tour/tree" // 引入 Go Tour 提供的 tree 包
)

// Walk 函数将二叉树 t 的所有值发送到通道 ch
// 并在所有值发送完毕后关闭通道。
func Walk(t *tree.Tree, ch chan int) {
    defer close(ch) // 确保通道在 Walk 函数返回时关闭

    var walk func(t *tree.Tree)
    walk = func(t *tree.Tree) {
        if t == nil {
            return
        }
        walk(t.Left)
        ch <- t.Value
        walk(t.Right)
    }
    walk(t)
}

// 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 {
        v1, ok1 := <-ch1
        v2, ok2 := <-ch2

        switch {
        case !ok1 && !ok2: // 两个通道都已关闭，且之前所有值都匹配
            return true
        case !ok1 || !ok2: // 一个通道关闭，另一个仍有值，表示不相等
            return false
        case v1 != v2:     // 值不匹配，表示不相等
            return false
        }
    }
}

func main() {
    // 测试两棵等价的树
    fmt.Println("tree.New(1) 和 tree.New(1) 是否等价:", Same(tree.New(1), tree.New(1))) // 预期输出: true
    // 测试两棵不等价的树
    fmt.Println("tree.New(1) 和 tree.New(2) 是否等价:", Same(tree.New(1), tree.New(2))) // 预期输出: false
    // 测试两棵结构相同但值不同的树 (例如，使用不同的种子生成)
    fmt.Println("tree.New(1) 和 tree.New(10) 是否等价:", Same(tree.New(1), tree.New(10))) // 预期输出: false
}