二叉树 Leetcode 872 叶子相似的树

2019-07-14  本文已影响0人  禾木清清

题目

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

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

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

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

提示:

给定的两颗树可能会有 1 到 100 个结点。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/leaf-similar-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

代码

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

class Solution(object):
    def leafSimilar(self, root1, root2):
        """
        :type root1: TreeNode
        :type root2: TreeNode
        :rtype: bool
        """
        if not root1 and root2:
            return False
        if not root2 and root1:
            return False
        if not root1 and not root2:
            return True
        
        def isLeaf(node):
            if node.left == None and node.right == None:
                return True
            else:
                return False
        
        def leafOrder(node):
            order = []
            if not node:
                return []
            if isLeaf(node):
                order.append(node.val)
            order.extend(leafOrder(node.left))
            order.extend(leafOrder(node.right))
            return order
        
        order1 = leafOrder(root1)
        order2 = leafOrder(root2)
        
        if len(order1) != len(order2):
            return False
        else:
            for i in range(len(order1)):
                if order1[i] != order2[i]:
                    return False
        return True
        

上一篇下一篇

猜你喜欢

热点阅读