leetcode112. 二叉树 路径总和 (python代码实现)

题目描述
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。

在这里插入图片描述
解题思路
注意:需要分情况讨论,当根节点为空时,不存在路径。
自顶向下遍历,每次遍历一条边的时候,sum减去这条变的值。

# Definition for a binary tree node.
#给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


class Solution:
    def hasPathSum(self, root: TreeNode, sum: int) -> bool:
        if not root: return False
        t = sum - root.val
        if not root.left and not root.right:
            return t == 0
        return self.hasPathSum(root.left, t) or self.hasPathSum(root.right, t)


n1 = TreeNode(4)
n2 = TreeNode(2)
n3 = TreeNode(7)
n4 = TreeNode(1)
n5 = TreeNode(3)
n1.left = n2
n1.right = n3
n2.left = n4
n2.right = n5
s = Solution()
result = s.hasPathSum(n1,10)
print(result)

猜你喜欢

转载自blog.csdn.net/u013075024/article/details/91366560
今日推荐