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

题目描述

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

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    void DFS(TreeNode* root,int expectNumber,int& sum,vector<vector<int> >& result,vector<int>& road)
    {
        sum += root->val;
        road.push_back(root->val);
        if(root->left == NULL && root->right == NULL && sum == expectNumber)
        {
            result.push_back(road);
        }
        if (root->left != NULL)
        {
            DFS(root->left,expectNumber,sum,result,road);
        }
        if(root->right != NULL)
        {
            DFS(root->right,expectNumber,sum,result,road);
        }
        sum -= root->val;
        road.pop_back();
    }
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        vector<vector<int> > result;
        if(root == NULL)
            return result;
        vector<int> road;
        int sum = 0;
        DFS(root,expectNumber,sum,result,road);
        return result;
    }
    
};