【每日一题】LeetCode 2236.判断根节点是否等于子节点之和（树、二叉树、递归）

【每日一题】LeetCode 2236.判断根节点是否等于子节点之和（树、二叉树、递归）

题目描述

给定一个由恰好三个节点组成的二叉树，其中包含一个根节点和两个子节点（左子节点和右子节点）。判断根节点的值是否等于其两个子节点值之和。

思路分析

此问题可以通过递归或迭代的方式解决。由于树的结构简单，我们可以直接访问根节点和其子节点的值进行比较。

1. 访问根节点：首先，我们需要访问到根节点。
2. 访问子节点：然后，访问根节点的左子节点和右子节点。
3. 比较值：最后，比较根节点的值是否等于其两个子节点的值之和。

注意，我们需要确保子节点不为空，因为空节点的值不能用于计算。

输入示例

• 示例 1：

  • 输入：root = [10,4,6]

  • 输出：true

  • 

  • 解释：根节点的值为10，左子节点的值为4，右子节点的值为6。因为10等于4+6，所以返回true。

• 示例 2：

  • 输入：root = [5,3,1]

  • 输出：false

  • 

  • 解释：根节点的值为5，左子节点的值为3，右子节点的值为1。因为5不等于3+1，所以返回false。

代码实现

class Solution {
    /**
     * 判断根节点是否等于子节点之和
     * @param root 二叉树的根节点
     * @return 如果根节点值等于两个子节点值之和，则返回true，否则返回false
     */
/**
 * 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 checkTree(TreeNode root) {
        if(root.right.val+root.left.val==root.val)
        return true;
        return false;
        
    }
}
}

拓展

假设输入的是一棵完全二叉树，树上有 n 个节点（n≥3）。要求判断除了叶子节点，是否每个节点的值都等于其左右儿子的节点值之和。

我们可以尝试用递归解决：

如果当前节点是叶子节点，直接返回 true，表示无需判断叶子。
否则，如果当前节点的节点值不等于其儿子的节点值之和，返回 false。
否则，递归左右儿子，如果它们都返回 true，那么当前节点返回 true，否则返回 false。

class Solution {
    public boolean checkTree(TreeNode root) {
        if(root.right==root.left){
            return true;
       
        }
        
       return root.val==root.left.val+root.right.val&&checkTree(root.left)&&checkTree(root.right);
        
    }
}