LeetCode---437. Path Sum III

原创于 2018-05-07 16:57:22 发布 · 187 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #二叉树

leetcode 专栏收录该内容

44 篇文章

订阅专栏

本文介绍了一种算法，用于寻找二叉树中所有节点数值和等于特定目标值的路径。这些路径可以从任意节点开始，并提供了Python实现示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

给出一个二叉树和一个目标值值，找到所有的路径，该路径上的节点数值和为目标值。路径不一定非要用根节点开始。举个例子：

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8
10
/ \
5 -3
/ \ \
3 2 11
/ \ \
3 -2 1

Return 3. The paths that sum to 8 are:

5 -> 3
5 -> 2 -> 1
-3 -> 11

Python题解

class Solution(object):
    def helper(self, root, target, so_far, cache):
        if root:
            complement = so_far + root.val - target
            if complement in cache:
                self.result += cache[complement]
            cache.setdefault(so_far + root.val, 0)
            cache[so_far + root.val] += 1
            self.helper(root.left, target, so_far + root.val, cache)
            self.helper(root.right, target, so_far + root.val, cache)
            cache[so_far + root.val] -= 1

    def pathSum(self, root, sum_val):
        """
        :type root: TreeNode
        :type sum_val: int
        :rtype: int
        """
        self.result = 0
        self.helper(root, sum_val, 0, {0: 1})
        return self.result