965. 单值二叉树(Python)

作者: 玖月晴 | 来源:发表于2019-06-01 17:06 被阅读0次

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

    题目

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

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

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

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

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

    相关文章

      网友评论

        本文标题:965. 单值二叉树(Python)

        本文链接:https://www.haomeiwen.com/subject/xjuatctx.html