leetcode —— 965. 单值二叉树

如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。

只有给定的树是单值二叉树时,才返回 true;否则返回 false。

输入:[1,1,1,1,1,null,1]
输出:true

——————————————
解题思路:对于当前节点,比较当前节点和左子节点的值是否相同,比较当前节点和右子节点的值是否相同。如果存在不同值,则直接返回False。如果值都相同,则使用递归函数比较其左子节点和右子节点。

其Python代码如下:

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

class Solution:
    def isUnivalTree(self, root: TreeNode) -> bool:
        if not root:
            return True
        if root.left and root.left.val!=root.val:  # 比较当前节点和左子节点
            return False
        if root.right and root.right.val!=root.val:  # 比较当前节点和右子节点
            return False
        return self.isUnivalTree(root.left) and self.isUnivalTree(root.right)  # 递归比较左子树和右子树

算法的时间复杂度为O(n),因为要比较每一个结点。空间复杂度为O(H),H表示树的高度,因为使用了递归遍历。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章