Golang Leetcode 872. Leaf-Similar Trees.go

2019-05-05 11:54:03 浏览数 (1)

版权声明:原创勿转 https://cloud.tencent.com/developer/article/1420472

思路

先递归找到所有的叶子节点,再循环判断

code

代码语言:javascript复制
type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}

func leaf(root *TreeNode, ret *[]int) {
	if root == nil {
		return
	}
	if root.Left == nil && root.Right == nil {
		*ret = append(*ret, root.Val)
	}
	leaf(root.Left, ret)
	leaf(root.Right, ret)
}
func leafSimilar(root1 *TreeNode, root2 *TreeNode) bool {
	a, b := []int{}, []int{}
	leaf(root1, &a)
	leaf(root2, &b)
	for k, v := range a {
		if b[k] != v {
			return false
		}
	}
	return true
}

0 人点赞