leetcode 437——路径总和III

题目描述:
  给定一个二叉树,它的每个结点都存放着一个整数值。
  找出路径和等于给定数值的路径总数。
  路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
  二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数。

示例:
在这里插入图片描述
解题思路:
首先我们想一下,如果是求 “从根节点出发,求根节点到任意节点和为sum的路径总数”,这个问题的解法很简单,我们进行递归求解即可,那么这里可以从任意节点出发,我们就可以把每个节点看做起点,对其进行递归求解,但这对每个节点都要进行两次递归。
两次递归代码实现如下:

class Solution {
public:
    int pathSum(TreeNode* root, int sum) {
        if (root == nullptr)
            return 0;
        
        return pathSumCore(root, sum) + 
               pathSum(root->left, sum) + 
               pathSum(root->right, sum);
    }
private:
    int pathSumCore(TreeNode* root, int sum)
    {
        if (root == nullptr)
            return 0;
        
        sum -= root->val;
        if (0 == sum) {
            return 1 + pathSumCore(root->left, sum) + 
                       pathSumCore(root->right, sum);
        }
        
        return pathSumCore(root->left, sum) + 
               pathSumCore(root->right, sum);
    }
};

下面再讲一种一次递归求解的方法:

  • 思想是遍历到某个节点时,检查该节点到根节点的路径上包含该节点的所有可能,如此每一个节点在检查可能的路径时都不会重复。
  • 比如假设有路径1,2,3,4,-4,5, sum=5,如果遍历到3时,只需要检查1,2,3中是否有包含3的解,发现2,3是一个解;
  • 遍历到5时只需检查包含5的解,包括 5,以及 4,-4,5 两个解。这样就没有重复检查。因为包含5的解在之前一定没被检查过。
  • 检查方式也是极其简单,使用vector保存遍历过程中的值,然后反向for循环这个vector求和即可

一次递归代码实现:

class Solution {
public:
    int pathSum(TreeNode* root, int sum) {
        vector<int> path;
        int res = 0;
        
        pathSumCore(root, sum, path, res);
        
        return res;
    }
private:
    void pathSumCore(TreeNode* root, int sum, vector<int>& path, int& res)
    {
        if (root == nullptr)
            return;
        
        path.push_back(root->val);
        
        int curSum = 0;
        for (int i = path.size()-1; i >= 0; --i)
        {
            curSum += path[i];
            if (curSum == sum)
                ++res;
        }
        
        pathSumCore(root->left, sum, path, res);
        pathSumCore(root->right, sum, path, res);
        
        path.pop_back();
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值