【每日一题】LeetCode 2236.判断根节点是否等于子节点之和(树、二叉树、递归)
题目描述
给定一个由恰好三个节点组成的二叉树,其中包含一个根节点和两个子节点(左子节点和右子节点)。判断根节点的值是否等于其两个子节点值之和。
思路分析
此问题可以通过递归或迭代的方式解决。由于树的结构简单,我们可以直接访问根节点和其子节点的值进行比较。
- 访问根节点:首先,我们需要访问到根节点。
- 访问子节点:然后,访问根节点的左子节点和右子节点。
- 比较值:最后,比较根节点的值是否等于其两个子节点的值之和。
注意,我们需要确保子节点不为空,因为空节点的值不能用于计算。
输入示例
-
示例 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);
}
}