二叉树中和为某一值的路径

来源:互联网 发布:网页软件游戏开发 编辑:程序博客网 时间:2024/05/29 12:51

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

----------------

/*struct TreeNode {    int val;    struct TreeNode *left;    struct TreeNode *right;    TreeNode(int x) :            val(x), left(NULL), right(NULL) {    }};*/class Solution {public:    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {        //利用先序遍历,用一队列暂存序列        vector<int> ele;        vector<vector<int> > res;        int s = 0;        preOrder(root, s, expectNumber, res, ele);        return res;              }private:    void preOrder(TreeNode* root, int &s, int &sum, vector<vector<int> > &res, vector<int> &ele){        if (!root)            return ;        s += root->val;        ele.push_back(root->val);        if (!root->left && !root->right){            if (s == sum){                res.push_back(ele);            }        }        if (root->left){            preOrder(root->left, s, sum, res, ele);            s -= ele.back();            ele.pop_back();//回一步        }        if (root->right){            preOrder(root->right, s, sum, res, ele);            s -= ele.back();            ele.pop_back();//回一步        }        return ;    }};


0 0
原创粉丝点击