给你一个 二叉树 的根结点 root,该二叉树由恰好 3 个结点组成:根结点、左子结点和右子结点。
如果根结点值等于两个子结点值之和,返回 true ,否则返回 false 。
示例 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 。
本题的重点主要在于了解TreeNode数据结构。
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
一个数的最小单元可划分为根节点,左节点,右节点。
定义root为一个具有双层三节点的TreeNode对象,可以通过root.val root.left.val root.right.val来获取各个节点的值。
对应本题题意,解法可写为:
class Solution:
def checkTree(self, root: Optional[TreeNode]) -> bool:
return root.val == root.left.val + root.right.val
灵神拓展:假设输入的是一棵完全二叉树(每个节点都有 0或 2 个儿子),树上有 n 个节点(n>=3)。要求判断除了叶子节点,是否每个节点的值都等于其左右儿子的节点值之和。
题解思路:首先判断该节点是否是叶子节点(无向下分支)
如果当前节点是叶子节点,直接返回 true,表示无需判断叶子。
否则,如果当前节点的节点值不等于其儿子的节点值之和,返回 false。
否则,递归左右儿子,如果它们都返回 true,那么当前节点返回 true,否则返回 false
class Solution:
def checkTree(self, root: Optional[TreeNode]) -> bool:
if root.left is root.right: # 递归边界:判断 root 是否为叶子节点
return True
return root.val == root.left.val + root.right.val and self.checkTree(root.left) and self.checkTree(root.right)
运用了递归的思想,代码中:如果root.left is root.right,即为左右节点都指向None,即为叶子节点,无需判断;再进行判断左右是否相等,并递归该节点的左右节点即可。
值得学习思考。