【刷题之路】二叉树中和为某一值的路径

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

问题难点在于, 为了确定某一条路径是否符合要求,需要走完整条路来确定是否要保留之前走的路径点

我们采取递归寻找的方法,将走过的路径全部保留,如果最后发现这条路并不是我们需要的,则将已经存入的路径点弹出,如果满足,则将整条路径返回

代码如下

/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
    vector<vector<int> > res; //定义全局变量,方便结果传递
    vector<int> temp;
    void help(TreeNode* root,int expectNumber){
        temp.push_back(root->val); //无论如何都将走过的节点值压入
        if(root->left==NULL && root->right==NULL && expectNumber-root->val==0) res.push_back(temp);  //如果满足条件,则返回这条路径
        else{
            if(root->left) help(root->left,expectNumber-root->val);  //否则继续递归
        if(root->right) help(root->right,expectNumber-root->val);
        }
        temp.pop_back(); //如果递归结束,res并没有返回,则从后开始弹出当前节点,重新走另一条路
    }
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        if(root) help(root,expectNumber);
        return res;
    }
    
};

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zyn2609530/article/details/51940521
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭