872. 叶子相似的树 简单

请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。

举个例子,如上图所示,给定一棵叶值序列为 (6, 7, 4, 9, 8) 的树。

如果有两棵二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。

如果给定的两个根结点分别为 root1 和 root2 的树是叶相似的,则返回 true;否则返回 false 。

示例 1:

输入:root1 = [3,5,1,6,2,9,8,null,null,7,4], root2 = [3,5,1,6,7,4,2,null,null,null,null,null,null,9,8]
输出:true

示例 2:

输入:root1 = [1], root2 = [1]
输出:true

示例 3:

输入:root1 = [1], root2 = [2]
输出:false

示例 4:

输入:root1 = [1,2], root2 = [2,2]
输出:true

示例 5:

输入:root1 = [1,2,3], root2 = [1,3,2]
输出:false

提示:

  • 给定的两棵树可能会有 1 到 200 个结点。
  • 给定的两棵树上的值介于 0 到 200 之间。

代码参考:

package main

func main() {
    // ok
}

func leafSimilar(root1 *TreeNode, root2 *TreeNode) bool {
    var leafs1, leafs2 []int
    traverse(root1, &leafs1)
    traverse(root2, &leafs2)
    if len(leafs1) != len(leafs2) {
        return false
    }
    for i, l1 := range leafs1 {
        if l1 != leafs2[i] {
            return false
        }
    }
    return true
}

func traverse(root *TreeNode, leafs *[]int) {
    if root == nil {
        return
    }
    traverse(root.Left, leafs)
    traverse(root.Right, leafs)
    if root.Left == nil && root.Right == nil {
        *leafs = append(*leafs, root.Val)
    }
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng