题目来源: 437.路径总和III
题目: 给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
思路: 本质在于是否将当前结点作为起点,若作为起点,则继续往前走,直至遍历到路径和为止。若不往前走,依次遍历左右子树,判断其是否作为起点,循环往复,直至找到路径和。
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)