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

最新推荐文章于 2019-04-13 04:02:39 发布

原创最新推荐文章于 2019-04-13 04:02:39 发布 · 405 阅读

0 ·

CC 4.0 BY-SA版权

程序员面试金典同时被 2 个专栏收录

21 篇文章

订阅专栏

树

7 篇文章

订阅专栏

本文介绍了一种算法，用于寻找一棵二叉树中所有节点值之和等于给定整数的路径。通过递归先序遍历的方法，实现路径搜索，并详细展示了如何使用临时存储队列来记录路径。

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

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

/*
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 ;
    }
};