LeetCode 437.路径总和

题目

给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

我的想法

递归中嵌套递归 一个递归负责遍历所有的节点（pathSum），一个递归负责检查从这个节点开始往下有多少路径符合要求（check）

class Solution:
    def pathSum(self, root: TreeNode, targetSum: int) -> int:
        def check(root,targetSum):
            if not root:
                return 0
            l = check(root.left , targetSum -root.val)
            r = check(root.right, targetSum - root.val)
            return l+r+(1 if root.val == targetSum else 0)
        if not root:
            return 0
        s = check(root,targetSum)
        l = self.pathSum(root.left,targetSum)
        r = self.pathSum(root.right,targetSum)
        return s+l+r

空间复杂度太高了

大佬想法

class Solution:
    def pathSum(self, root: TreeNode, targetSum: int) -> int:
        d = {}
        def check(root,acc,targetSum):
            ans = 0 
            if not root:
                return 0 
            acc += root.val
            if acc == targetSum:
                ans +=1
            if acc - targetSum in d:
                ans += d[acc - targetSum]
            if acc in d:
                d[acc] +=1
            else:
                d[acc] =1
            ans += check(root.left,acc,targetSum)+check(root.right,acc,targetSum)
            d[acc] = d[acc] - 1;
            return ans
        return check(root,0,targetSum)

用map来保存累加和，以及累加和出现的次数，当 累加和-目标值 在map中存在，则证明满足条件 将次数加起来 是LeetCode 560. 和为 K 的子数组 的进阶版，只不过从数组变为二叉树（相当于很多条数组啦（10 5 3 3 ，10 5 3 -2 ，10 5 2 1…））