二叉树 Leetcode 965 单值二叉树

2019-07-14  本文已影响0人  禾木清清

题目

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

只有给定的树是单值二叉树时,才返回 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

先获取每个节点放入一个list。最后使用set函数查看是否只有一个值。返回true or false。

代码

# 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
        """
        
        def get_nodes(root):
        
            if not root:
                return []
        
            res = [root.val]
        
            res.extend(get_nodes(root.left))
            res.extend(get_nodes(root.right))
            return res
        
        values = get_nodes(root)
        if len(set(values)) == 1:
            return True
        else:
            return False
     

上一篇下一篇

猜你喜欢

热点阅读