【题目】
如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true;否则返回 false。
示例 1:
输入:[1,1,1,1,1,null,1]
输出:true
示例 2:
输入:[2,2,2,5,2]
输出:false
提示:
给定树的节点数范围是 [1, 100]。
每个节点的值都是整数,范围为 [0, 99] 。
【代码】
【方法1:广度优先】
class Solution:
def isUnivalTree(self, root: TreeNode) -> bool:
ans=set()
que=[root]
while que:
node=que.pop(0)
ans.add(node.val)
if node.left:
que.append(node.left)
if node.right:
que.append(node.right)
return len(ans)==1
【方法2:深度优先】
class Solution:
def visit(self,root):
if not root:
return
self.ans.add(root.val)
self.visit(root.left)
self.visit(root.right)
def isUnivalTree(self, root: TreeNode) -> bool:
self.ans=set()
self.visit(root)
return len(self.ans)==1
【方法3:深度优先优化版本(推荐)】
class Solution(object):
def isUnivalTree(self, root):
left_correct = (not root.left or root.val == root.left.val
and self.isUnivalTree(root.left))
right_correct = (not root.right or root.val == root.right.val
and self.isUnivalTree(root.right))
return left_correct and right_correct