(剑指Offer)二叉树中和为某一值的路径

题目描述

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

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> >result;
    void recursion(TreeNode* root,vector<int>& path,int expectNumber,int cnt)
    {
        cnt+=root->val;
        path.push_back(root->val);
        if(root->left==nullptr&&root->right==nullptr)
            if(cnt==expectNumber)
            {
                result.push_back(path);
                path.pop_back();
                return ;
            }
        if(root->left!=nullptr)
            recursion(root->left,path,expectNumber,cnt);
        if(root->right!=nullptr)
            recursion(root->right,path,expectNumber,cnt);
        path.pop_back();
        return ;
    }
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber)
    {
        vector<int> path;
        if(root==nullptr) return result;
        int cnt=0;
        recursion(root,path,expectNumber,cnt);
        return result;
    }
};

• vector<int> &path：使用引用时，所有函数使用一个数组，函数结束时应弹出值。如果不用引用就不用弹出。
• 递归一遍，值存入数组中，到底->如果得到的值相等存入vector。只在最底层叶子结点进行判断。

2.在原函数上递归，改变expectNumber的值(仿照牛客网上的写的)

class Solution
{
public:
    vector<int> path;
    vector<vector<int> > result;
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber)
    {
        if(root==nullptr)
            return result;
        path.push_back(root->val);
        if(expectNumber-(root->val)==0&&(root->left==nullptr&&root->right==nullptr))
            result.push_back(tmp);
        if(root->left!=nullptr) FindPath(root->left,expectNumber-(root->val));
        if(root->right!=nullptr) FindPath(root->right,expectNumber-(root->val));
        if(path.size()) path.pop_back();
        return result;
    }
};