1. 问题描述
如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true
;否则返回 false
。
示例 1:
输入:[1,1,1,1,1,null,1] 输出:true
示例 2:
输入:[2,2,2,5,2] 输出:false
提示:
- 给定树的节点数范围是
[1, 100]
。 - 每个节点的值都是整数,范围为
[0, 99]
。
2. 思路与算法
基本的树遍历的问题。
以某个节点为根的子树是单值二叉树,必须同时满足以下条件:
- 它的左子树(如果存在的话)是单值二叉树
- 它的右子树(如果存在的话)是单值二叉树
- 左子树(如果存在的话)的值、右子树(如果存在的话)的值等于它自身的值
据此,可以以递归的方式来实现。
3. 代码实现
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def isUnivalTree(self, root: TreeNode) -> bool:
def dfs(r: TreeNode)->bool:
isLeftUnival = (r.left == None) or (r.left.val == r.val and dfs(r.left))
isRightUnival = (r.right == None) or (r.right.val == r.val and dfs(r.right))
return (r==None) or (isLeftUnival and isRightUnival)
return dfs(root)
执行用时:32 ms, 在所有 Python3 提交中击败了93.20%的用户
内存消耗:15 MB, 在所有 Python3 提交中击败了12.87%的用户
回到总目录:Leetcode每日一题总目录(动态更新。。。)