leetcode437——Path SumⅢ

最新推荐文章于 2021-12-04 10:56:23 发布

tzyshiwolaogongya

最新推荐文章于 2021-12-04 10:56:23 发布

阅读量133

点赞数

分类专栏：搜索-dfs

本文链接：https://blog.youkuaiyun.com/tzyshiwolaogongya/article/details/105112609

版权

搜索-dfs 专栏收录该内容

30 篇文章

订阅专栏

本文探讨了一种算法，用于查找二叉树中所有路径和等于特定目标值的路径数量，路径可以从任意节点开始，且不必以叶子节点结束。通过深度优先搜索(DFS)策略实现，针对每个节点作为起点进行递归计数，并调整路径和，适用于节点值可能为负数的情况。

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

题目大意：找出二叉树中路径和等于目标和的路径个数，节点为整数（可能为负），路径起点不一定为根节点，终点不一定为叶节点，但是路径方向始终向下

分析：二叉树dfs。相比leetcode112、113要注意三点：节点数值可能为负，路径起点可以任选，路径终点不一定是叶节点。

因为路径起点任选，所以要以树中每一个节点为起点dfs，也就是用countHelper统计以该点为起点的路径条数，然后递归统计左右节点为起点的路径条数。

countHelper统计路径条数：注意由于节点值可能为负，所以此时不能return，还要继续递归下去。

由于路径终点不一定是叶节点，所以符合目标的条件比leetcode112、113少了一个，不用判断节点左右孩子是否为空了。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    int ans = 0;
public:
    int pathSum(TreeNode* root, int sum) {
        if(!root) return 0;
        countHelper(root,sum);
        pathSum(root->left,sum);
        pathSum(root->right,sum);
        return ans;
    }
    void countHelper(TreeNode* root, int sum){
        if(!root) return;
        if(root->val == sum){
            ans++;
        }
        countHelper(root->left,sum - root->val);
        countHelper(root->right,sum - root->val);
    }
};