equivalent_binary_trees.go

package main

import (
    "code.google.com/p/go-tour/tree"
    "fmt"
)

func Walk(t *tree.Tree, ch chan int) {
    WalkIter(t, ch)
    close(ch)
}

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

// 二分木の比較
func Same(t1, t2 *tree.Tree) bool {
    c1 := make(chan int, 10)
    c2 := make(chan int, 10)
    go Walk(t1, c1)
    go Walk(t2, c2)
    for i := range c1 {
        if i != <-c2 {
            return false
        }
    }
    return true
}

func main() {
    // fmt.Println(tree.New(1))
    // ((((1 (2)) 3 (4)) 5 ((6) 7 ((8) 9))) 10)
    // 空リストを表示してもpanicにはならない
    // 長さ10のチャンネルを作成
    ch := make(chan int, 10)
    go Walk(tree.New(1), ch)
    for i := range ch {
        fmt.Println(i)
    }

    fmt.Println(Same(tree.New(1), tree.New(1)))
    fmt.Println(Same(tree.New(1), tree.New(2)))
}