Go Tour Exercise: Equivalent Binary Trees

前端 未结 23 1793
攒了一身酷
攒了一身酷 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:19

    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 Walk(t *tree.Tree, ch chan int) {
        walkRecursive(t, ch)
        close(ch)
    }
    
    func walkRecursive(t *tree.Tree, ch chan int) {
        if t == nil {
            return
        }
        walkRecursive(t.Left, ch)
        ch <- t.Value
        walkRecursive(t.Right, 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 {
            v1, ok1 := <-ch1
            v2, ok2 := <-ch2
            if ok1 != ok2 {
                return false
            }
            if !ok1 {
                return true
            }
            if v1 != v2 {
                return false
            }
        }
    }
    
    func main() {
        fmt.Println(Same(tree.New(1), tree.New(2)))
    }
    
    

提交回复
热议问题