Go Tour Exercise: Equivalent Binary Trees

前端 未结 23 1846
攒了一身酷
攒了一身酷 2020-12-12 23:39

I am trying to solve equivalent binary trees exercise on go tour. Here is what I did;

package main

import \"tour/tree\"
import \"fmt\"

// Walk walks the tr         


        
23条回答
  •  夕颜
    夕颜 (楼主)
    2020-12-13 00:30

    My version

    package main
    
    
    import (
        "fmt"
        "golang.org/x/tour/tree"
    )
    
    // Walk walks the tree t sending all values
    // from the tree to the channel ch.
    func WalkRec(t *tree.Tree, ch chan int) {
        if t == nil {
            return
        }
        WalkRec(t.Left, ch)
        ch <- t.Value
        WalkRec(t.Right, ch)
    }
    
    func Walk(t *tree.Tree, ch chan int) {
        WalkRec(t, ch)
        close(ch)
    }
    
    // Same determines whether the trees
    // t1 and t2 contain the same values.
    func Same(t1, t2 *tree.Tree) bool {
        ch1 := make(chan int)
        ch2 := make(chan int)
        go Walk(t1, ch1)
        go Walk(t2, ch2)
    
        for {
            x, okx := <-ch1
            y, oky := <-ch2
            switch {
            case okx != oky:
                return false
            case x != y:
                return false
            case okx == oky && okx == false:
                return true
            }
    
        }
    
    }
    
    func main() {
        ch := make(chan int)
        go Walk(tree.New(1), ch)
        fmt.Println(Same(tree.New(1), tree.New(1)))
        fmt.Println(Same(tree.New(2), tree.New(1)))
        fmt.Println(Same(tree.New(1), tree.New(2)))
    }
    

提交回复
热议问题