题目
如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true
;否则返回 false
。
代码
# 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 isUnivalTree(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
uni_value = root.val
s = [root]
while s:
cur = s.pop()
if cur.val != uni_value:
return False
if cur.right:
s.append(cur.right)
if cur.left:
s.append(cur.left)
return True