刷题使我快乐,满脸开心.jpg
- 来源:力扣(LeetCode)
- 链接:https://leetcode.cn/problems/univalued-binary-tree/
- 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目
如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true
;否则返回 false
。
示例 1:
输入:[1,1,1,1,1,null,1]
输出:true
示例 2:
输入:[2,2,2,5,2]
输出:false
提示:
- 给定树的节点数范围是
[1, 100]
。 - 每个节点的值都是整数,范围为
[0, 99]
。
思路
无论何种顺序的遍历均可,只是在退出条件上要注意下,别少走分支了就好
代码
func checkNode(now *TreeNode) bool {
if now.Left != nil {
if now.Val != now.Left.Val {
return false
}
isOk := checkNode(now.Left)
if !isOk {
return false
}
}
if now.Right != nil {
if now.Val != now.Right.Val {
return false
}
return checkNode(now.Right)
}
return true
}
func isUnivalTree(root *TreeNode) bool {
if root == nil {
return true
}
return checkNode(root)
}