【树-简单】965. 单值二叉树（复习方法3）

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

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