【Leetcode_总结】872. 叶子相似的树 - python

Q:

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

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

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

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

---

链接:https://leetcode-cn.com/problems/leaf-similar-trees/

思路:深度优先遍历叶子节点，判断叶子节点的值是否相同

代码：

递归：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def leafSimilar(self, root1, root2):
        """
        :type root1: TreeNode
        :type root2: TreeNode
        :rtype: bool
        """
        
        return self.getLeafs(root1)==self.getLeafs(root2)
        
    def getLeafs(self,nd):
        if nd is None:
            return []
        if not (nd.left or nd.right):
            return [nd.val]
        return self.getLeafs(nd.left)+self.getLeafs(nd.right)

栈：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def leafSimilar(self, root1, root2):
        """
        :type root1: TreeNode
        :type root2: TreeNode
        :rtype: bool
        """
        
        return self.leaflist(root1)==self.leaflist(root2)
        
    def leaflist(self,root):
        res=''
        if not root:
            return res
        s=[root]
        while s:
            node=s.pop()
            if node.left:
                s.append(node.left)
            if node.right:
                s.append(node.right)
            if not node.left and not node.right:
                res+=str(node.val)
        return res