每日一题 2023.08.23 判断根结点是否等于子结点之和

给你一个 二叉树 的根结点 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,即为叶子节点,无需判断;再进行判断左右是否相等,并递归该节点的左右节点即可。

值得学习思考。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值