判断树路径上的和是否为某个数

这是一篇关于利用深度优先搜索(DFS)解决二叉树路径和问题的博客。作者通过牛客网的剑指Offer题目,分享了如何在C++中实现判断从根节点到叶节点的树路径和是否等于给定数值。代码实现中,重点在于DFS函数的设计,包括参数传递和路径值的处理。

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

题目来源:牛客网-剑指offer

题目描述:

输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

自己总结的github地址:
https://github.com/Rayholle/LeetCode-Newcoder/blob/master/Tree/判断树路径上的和是否为某个数.cpp
思路分析:路径问题从树的根结点到叶结点,所以需要深度优先遍历(dfs),在递归过程中发现当前路径和为给定的数时,将该路径存入vector中,再遍历其左子树与右子树。

细节:
1、dfs函数参数:
需要树结点,二维数组(最终返回值),一维数组(路径trace),当前需要值。
所以我们声明dfs函数为:注意res和trace用引用传递
void dfs(TreeNode* root, vector<vector<int>>& res, vector<int>& trace, int s);
2、关于dfs的函数定义:
需要先将该结点的val存储在trace,在函数定义结尾,需要将该值从trace中pop出去(当前递归层结束)。

void dfs(TreeNode* root, vector<vector<int>>& res, vector<int>& trace, int s){
    trace.push_back(root->val);
    if(!root->left&&!root->right){//当root为叶结点时,题目要求路径为从根到叶结点
        if(s==root->val)
            res.push_back(trace);
    if(root->left)
        dfs(root->left, res, trace, s-root->val);
    if(root->right) 
        dfs(root->right, res, trace, s-root->val); 
    trace.pop_back(root->val);
} 

接下来是完整代码:

vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
    vector<int> trace;
    vector<vector<int>> ret;
    if(root)
        dfs(root, expectNumber, trace, ret);
    return ret;
}
void dfs(TreeNode* root,int S, vector<int>& trace, vector<vector<int>>& ret){
    trace.push_back(root->val);
    //判断是否为叶子节点
    if(!root->left && !root->right){
        if(S == root->val)
            ret.push_back(trace);
    }
    if(root->left)
        dfs(root->left, S-root->val, trace, ret);
    if(root->right)
        dfs(root->right, S-root->val, trace, ret);
    trace.pop_back();
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值