题目描述
难度:【简单】
标签:【二叉树】
如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true;否则返回 false。
示例
示例 1
输入:[1,1,1,1,1,null,1]
输出:true
示例 2
输入:[2,2,2,5,2]
输出:false
题目大意
所有二叉树的节点的值都一样,返回 true,否则返回 false 。
解题
站在其中一个节点(a)的角度:
- a 的左右2个子节点的值相等,且 a 的值等于子节点的值
- 如果 a 的某一边的节点是null,那么忽略,只考虑另一边的结果
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public boolean isUnivalTree(TreeNode root) {
if (root == null) {
return false;
}
// 检查左子节点
boolean checkLeft = (root.left == null) || ((root.left.val == root.val) && isUnivalTree(root.left));
// 检查右子节点
boolean checkRight = (root.right == null) || ((root.right.val == root.val) && isUnivalTree(root.right));
return checkLeft && checkRight;
}
}
明确函数的「定义」是什么,然后相信这个定义,利用这个定义推导最终结果,绝不要跳入递归的细节。