【树-简单】872. 叶子相似的树

最新推荐文章于 2022-01-07 17:32:14 发布

原创最新推荐文章于 2022-01-07 17:32:14 发布 · 88 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

本文介绍了一种判断两棵二叉树是否为叶相似的方法。通过递归或迭代方式收集每棵树的所有叶子节点值，并比较两者的序列是否一致。提供三种实现方案并附带示例。

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

举个例子，如上图所示，给定一棵叶值序列为 (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 之间。

【代码】
【方法1 递归】
在这里插入图片描述

class Solution:
    def postOrder(self,root,flag):
        if not root:
            return
        self.postOrder(root.left,flag)
        if not root.left and not root.right:
            self.ans[flag].append(root.val)
        self.postOrder(root.right,flag)
    def leafSimilar(self, root1: TreeNode, root2: TreeNode) -> bool:
        self.ans=[[],[]]
        self.postOrder(root1,0)
        self.postOrder(root2,1)
        print(self.ans)
        return self.ans[0]==self.ans[1]

【方法2：迭代】
在这里插入图片描述

class Solution:
    def deal(self,node,flag):
        stack=[]
        while stack or node:
            while node:
                stack.append(node)
                node=node.left
            node=stack.pop()
            if not node.left and not node.right:
                self.ans[flag].append(node.val)
            node=node.right
    def leafSimilar(self, root1: TreeNode, root2: TreeNode) -> bool:
        if not root1 and not root2:
            return True
        if not root1 or not root2:
            return False
        self.ans=[[],[]]
        self.deal(root1,0)
        self.deal(root2,1)
        return self.ans[0]==self.ans[1]

【方法3 】
在这里插入图片描述

class Solution:
    def leafSimilar(self, root1: TreeNode, root2: TreeNode) -> bool:
        def dfs(node: TreeNode):
            if not node.left and not node.right:
                yield node.val
            else:
                if node.left:
                    yield from dfs(node.left)
                if node.right:
                    yield from dfs(node.right)
        
        seq1 = list(dfs(root1)) if root1 else list()
        seq2 = list(dfs(root2)) if root2 else list()
        return seq1 == seq2