刷题使我快乐,满脸开心.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)
}