【Python】【难度：简单】Leetcode 965. 单值二叉树

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

只有给定的树是单值二叉树时，才返回 true；否则返回 false。

 

示例 1：

输入：[1,1,1,1,1,null,1]
输出：true
示例 2：

输入：[2,2,2,5,2]
输出：false
 

提示：

给定树的节点数范围是 [1, 100]。
每个节点的值都是整数，范围为 [0, 99] 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/univalued-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

# 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
        """
        res=set()
        queue=[root]

        while queue:
            nxt=[]
            var=set()

            for node in queue:
                var.add(node.val)
                if node.left:
                    nxt.append(node.left)
                if node.right:
                    nxt.append(node.right)

            if len(var)==1:
                res.add(var.pop())
            elif len(var)>1:
                return False
            queue=nxt

        if len(res)==1:
            return True
        else:
            return False

 

执行结果：

通过

显示详情

执行用时 :16 ms, 在所有 Python 提交中击败了91.01%的用户

内存消耗 :12.8 MB, 在所有 Python 提交中击败了100.00%的用户