面试题 04.12. 求和路径

最新推荐文章于 2024-03-18 11:16:01 发布

原创最新推荐文章于 2024-03-18 11:16:01 发布 · 169 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #动态规划

程序员面试金典（第 6 版）专栏收录该内容

38 篇文章

订阅专栏

该博客讨论了一种在二叉树中寻找路径和等于给定值的算法问题。提供了一个解决方案，通过递归方法计算路径数量。示例展示了如何在包含负数的二叉树中找到和为22的路径，如[5,4,11,2]、[5,8,4,5]和[4,11,7]。

题

给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

示例:
给定如下二叉树，以及目标和 sum = 22，

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \    / \
    7    2  5   1

3
解释：和为 22 的路径有：[5,4,11,2], [5,8,4,5], [4,11,7]

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/paths-with-sum-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

注意会有负数即有可能会有 -1 -1+1-1 最后还是-1的路径基本上不能用剪枝

代码


class Solution
{
public:
    int pathSum(TreeNode *root, int sum)
    {
        if (!root)
        {
            return 0;
        }
        return have(root, 0, sum) + pathSum(root->left, sum) + pathSum(root->right, sum);
    }

    int have(TreeNode *root,int current, int sum)
    {
        if (!root)
        {
            return 0;
        }
        int value = current + root->val;
        int ans = 0;
        if (value == sum)
        {
            ans++;
        }
        return ans+ have(root->left, value, sum) + have(root->right, value, sum);
    }
};