【LeetCode】965. 单值二叉树

题目描述

难度：【简单】
标签：【二叉树】

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

只有给定的树是单值二叉树时，才返回 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; 
    }
}

明确函数的「定义」是什么，然后相信这个定义，利用这个定义推导最终结果，绝不要跳入递归的细节。