437.路径总和III

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

法一:深度优先遍历,递归

  • rootSum(p,val)计算节点p向下且路径和为val的路径和
  • rootSum(p,val)=rootSum(p->left,val-pval)+rootSum(p->right,val-pval)
  • 遍历二叉树所有节点的rootSum并求和
class Solution {
public:
    int rootSum(TreeNode* node,long target){
        if(!node){
            return 0;
        }
        int ret=0;
        int val=node->val;
        if(val==target){
            ++ret;
        }
        ret+=rootSum(node->left,target-val);
        ret+=rootSum(node->right,target-val);
        return ret;
    }

    int pathSum(TreeNode* root, int targetSum) {
        if(!root){
            return 0;
        }
        int ret=0;
        ret=rootSum(root,targetSum);
        ret+=pathSum(root->left,targetSum);
        ret+=pathSum(root->right,targetSum);
        return ret;
    }
};

法二:前缀和+深度优先遍历

  • 根节点root到node的前缀和sum
  • 已保存的前缀和中查找是否存在curr-target
class Solution {
public:
    unordered_map<long long,int>prefix;

    int dfs(TreeNode* node,long long sum,int target){
        if(!node) return 0;

        int ret=0;
        sum+=node->val;
        if(prefix.count(sum-target)){
            ret=prefix[sum-target];
        }

        ++prefix[sum];
        ret+=dfs(node->left,sum,target);
        ret+=dfs(node->right,sum,target);
        --prefix[sum];

        return ret;
    }
    int pathSum(TreeNode* root, int targetSum) {
        prefix[0]=1;
        return dfs(root,0,targetSum);
    }    
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值