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

题目描述

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

递归代码：

class Solution {
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        vector<vector<int> >ans;
        vector<int>res;
        if(root==NULL)return ans;
        path(root,ans,res,expectNumber);
        return ans;
    }
    void path(TreeNode *root,vector<vector<int> >&ans,vector<int>&res,int num){
        if(root){
            res.push_back(root->val);
            if(!root->left&&!root->right&&root->val==num)
                ans.push_back(res);
            if(root->left)
                path(root->left,ans,res,num-root->val);
            if(root->right)
                path(root->right,ans,res,num-root->val);
            res.erase(res.end()-1);
        }
    }
};

非递归代码：

class Solution {
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        vector<vector<int> >ans;
        vector<int>res;
        if(root==NULL)return ans;
        stack<TreeNode *>s;
        TreeNode *cur=root;
        TreeNode *last=NULL;
        int sum=0;
        while(cur||!s.empty()){
            if(cur){
                s.push(cur);
                res.push_back(cur->val);
                sum+=cur->val;
                if(!cur->left&&!cur->right&&sum==expectNumber)ans.push_back(res);
                cur=cur->left;
            }else{
                TreeNode *temp=s.top();
                if(temp->right&&temp->right!=last)cur=temp->right;//转向右子树
                else{
                    s.pop();
                    last=temp;//只要记住：last一直指向最近一个弹出来的结点
                    sum-=temp->val;
                    res.pop_back();
                }
            }
        }
        return ans;
    }
};

递归代码还是不熟，一定要多写。