437.路径总和Ⅲ

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

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

示例 1:

在这里插入图片描述
输入:root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出:3
解释:和等于 8 的路径有 3 条,如图所示。

示例 2:

输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:3

解题思路

题目要求的是,可以从任意结点开始计算路径的和,所以首先用dfs思想写一个以当前结点为起点的处理函数rootSum,然后在函数pathSum中通过递归,实现对每个结点的处理

rootSum函数
采用递归方法,首先检查当前节点的值是否等于目标和,如果是,则表明从当前节点到自身构成了一条满足条件的路径,因此结果计数加1。然后,分别对左子树和右子树递归调用自身,每次调用时将目标和减去当前节点的值,以寻找剩余路径上需要累积的值。最后,返回所有这些情况下的路径总数。

pathSum函数
首先调用 rootSum 函数,从根节点开始寻找满足条件的路径。然后,递归地对根节点的左子树和右子树分别调用 pathSum,累加这两个子树中所有满足条件的路径数量。这样做确保了树中的每一个节点都被考虑到了至少一次,从而找到了所有可能的满足条件的路径。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left),
 * right(right) {}
 * };
 */
class Solution {
    int rootSum(TreeNode* node, long targetSum) {//注意这里使用long类型,不然有可能溢出
        if (node == nullptr) {
            return 0;
        }
        long ret = 0;
        if (node->val == targetSum) {
            ret++;
        }
		//通过targetSum - node->val,可以判断下一个结点的值,是否满足要求
        ret += rootSum(node->left, targetSum - node->val);
        ret += rootSum(node->right, targetSum - node->val);
        return ret;
    }

public:
    int pathSum(TreeNode* root, int targetSum) {
        if (root == nullptr) {
            return 0;
        }
        int ret = 0;
        ret = rootSum(root, targetSum); // 以根节点为起点

        // 然后是左右子树
        ret = ret + pathSum(root->right, targetSum);
        ret = ret + pathSum(root->left, targetSum);
        return ret;
    }
};
  • 3
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

青石横刀策马

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值