965. 单值二叉树（Python）

更多精彩内容，请关注【力扣简单题】。

题目

难度：★☆☆☆☆
类型：二叉树

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

只有给定的树是单值二叉树时，才返回 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

如果对上述中序遍历的一行代码有疑问，请移步【刷力扣二叉树题之前，你需要知道的】。

如有疑问或建议，欢迎评论区留言~