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

[guanpengchn]

题目描述

输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

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