力扣437.路径总和III(python)

题目来源: 437.路径总和III
题目: 给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
力扣437.路径总和III(python)_第1张图片
力扣437.路径总和III(python)_第2张图片
思路: 本质在于是否将当前结点作为起点,若作为起点,则继续往前走,直至遍历到路径和为止。若不往前走,依次遍历左右子树,判断其是否作为起点,循环往复,直至找到路径和。
python代码:

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def pathSum(self, root, targetSum):
        """
        :type root: TreeNode
        :type targetSum: int
        :rtype: int
        """
        if not root:
            return 0
        return self.pathSum(root.left, targetSum) + self.pathSum(root.right, targetSum) + self.pathsumwithroot(root, targetSum)
    def pathsumwithroot(self, root, targetSum):
        if not root:
            return 0
        if root.val == targetSum:
            return 1 + self.pathsumwithroot(root.left, 0) + self.pathsumwithroot(root.right, 0)
        return self.pathsumwithroot(root.left, targetSum - root.val) + self.pathsumwithroot(root.right, targetSum - root.val)

力扣437.路径总和III(python)_第3张图片
参考链接: leetcode(62)_437_medium_路径总和 III_python

你可能感兴趣的:(力扣刷题笔记,leetcode,算法,深度优先)