ITPub博客

首页 > 应用开发 > Python > 刷题系类 - Python判断二叉树是否存在一条路径满足和值要求

刷题系类 - Python判断二叉树是否存在一条路径满足和值要求

原创 Python 作者:张国平 时间:2020-01-20 11:10:28 0 删除 编辑

继续刷题,判断二叉树是否存在一条路径满足和值要求。

这个名称描述很复杂,其实很简单。如下图,判断是否有一个从根节点到叶子节点路径,和值为8, 就是1 -> 2-> 5。

这个用递归来做很简单,就是判断输入节点的值和输入值是否一样,如果是而且当前节点为叶子节点,返回True,如果当前节点为空,返回False; 如果都不是,则递归到输入节点子节点,并且输入值为sum减去当前节点。

这里使用or 来判单,只有有一组路径复合就可以返回True

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def hasPathSum(self, root: TreeNode, sum: int) -> bool:
        if root == None:
            return False
        else:
            if root.left == None and root.right == None and root.val == sum:
                return True
            else:
                return self.hasPathSum(root.left,sum-root.val) or self.hasPathSum(root.right,sum-root.val)


来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/22259926/viewspace-2674079/ ,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论
SAP 金融风险管理产品专家

注册时间:2009-08-05

  • 博文量
    229
  • 访问量
    635939