更多精彩内容,请关注【力扣简单题】。
题目
难度:★☆☆☆☆
类型:二叉树
如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true;否则返回 false。
提示
给定树的节点数范围是 [1, 100]。
每个节点的值都是整数,范围为 [0, 99] 。
示例
示例1
示例1输入:[1,1,1,1,1,null,1]
输出:true
示例2
示例2输入:[2,2,2,5,2]
输出:false
解答
我们可以先对二叉树进行中序遍历(或其他顺序),把所有结点放在一个列表中,如果去重后只有一个元素,则二叉树为单值二叉树。
class Solution:
def isUnivalTree(self, root):
"""
:param root: TreeNode
:return: bool
"""
vals = lambda n: vals(n.left) + [n.val] + vals(n.right) if n else []
return len(set(vals(root))) == 1
如果对上述中序遍历的一行代码有疑问,请移步【刷力扣二叉树题之前,你需要知道的】。
如有疑问或建议,欢迎评论区留言~
网友评论