113. 路径总和 (剑指 Offer 34. 二叉树中和为某一值的路径)(回溯算法)

10 篇文章 0 订阅
10 篇文章 0 订阅

给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。

叶子节点 是指没有子节点的节点。

示例 1:
在这里插入图片描述
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]

示例 2:
在这里插入图片描述
输入:root = [1,2,3], targetSum = 5
输出:[]

示例 3:
输入:root = [1,2], targetSum = 0
输出:[]

提示:

树中节点总数在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000

这道题其实比较好做的,但是需要打破前几道回溯算法那样的思路,开始就说过回溯类的题目都可以写成N叉树的形式,这个正好又是二叉树,所以就很简单了;

开始想到了for循环是不用写的,只需要分别递归左右子树就可以了,
且遍历到叶子节点时,如果路径所有节点和和targetSum相等,这条路径就是答案;

但是写出来发现了一个问题,当判断出来叶子节点后,此时的路径path并没有将叶子节点加进去,sum也一样,所以我就单独在结束时又加了一遍并回溯结束,这样就可以了;

而且当root节点为空时也需要判断,这个要写在判断叶子节点下面;
写在下面也面临一个问题就是叶子节点不能直接判断,所以我又通过一个函数来判断叶子节点;

详细注释代码如下:

/**
 * 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 {
public:
    vector<vector<int>> ans;
    vector<int> path;
    int sum = 0;
    void backTracking(TreeNode *root, int targetSum) {
        //判断是否为叶子节点
        if (isLeafnode(root)) {
            //为叶子节点,此时sum还没有加上叶子节点的值,所以先加上叶子节点val看看是否和
            //targetSum相等,如果相等,则将最后一个节点就是叶子节点压入path,此时path就是
            //答案之一,把path加入ans,然后回溯将这个叶子节点再从path里拿出来
            if (sum + root -> val == targetSum) {
                path.push_back(root -> val);
                ans.push_back(path);
                path.pop_back();
            }
            return ;
        }
        //这个判断是为了保证节点不为空
        if (!root) return ;
        sum += root -> val;
        path.push_back(root -> val);
        backTracking(root -> left, targetSum);//向下递归左子树
        backTracking(root -> right, targetSum);//向下递归右子树
        path.pop_back();//回溯
        sum -= root -> val;//回溯
    }
    //判断节点是否为叶子节点
    bool isLeafnode(TreeNode* root) {
        if (!root) return false;
        if (!root -> left && !root -> right) return true;
        return false;
    }
    vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
        if (!root) return ans;
        backTracking(root, targetSum);
        return ans;
    }
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

YXXYX

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

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

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

打赏作者

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

抵扣说明:

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

余额充值