二叉树中和为某一个值的路径(剑指offer-34)

1、题目描述

    输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

2、解题思路

    可以使用一个栈来保存二叉树的路径,使用先序遍历的方式遍历二叉树。当遍历到一个节点时,将这个节点压入栈中。接着判断这个节点是否是叶子节点,如果是,计算栈中所有节点的值是否为目标整数,接着返回父节点,将叶子节点值从栈中弹出。如果不是叶子节点,当左子树为真时,则递归遍历左子树,如果右子树为真时,递归遍历右子树。递归代码如下:

    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        vector<vector<int>> output;
        if(root==nullptr)                 //由于没有考虑空指针的情况,很容易发生段错误
            return output;
        vector<int> path;                 //使用一个vector来保存路径
        int cur_sum = 0;                  //用一个变量来保存当前路径和
        find(root,expectNumber,cur_sum,path,output);
        return output;
    }

    void find(TreeNode *p,int expectNumber, int &cur_sum,vector<int> &path,vector<vector<int>> &output){
        cur_sum+=p->val;
        path.push_back(p->val);
        if(p->left==nullptr&&p->right==nullptr&&cur_sum==expectNumber){
            output.push_back(path);
        }
        if(p->left)
            find(p->left,expectNumber,cur_sum,path,output);
        if(p->right)
            find(p->right,expectNumber,cur_sum,path,output);
        cur_sum-=p->val;
        path.pop_back();                  //返回父节点时路径中弹出当前节点值
        return;
    }

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值