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

题目描述

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

/*
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<vector<int> > ivecvec;
        if (NULL == root) {
            return ivecvec;
        }
        //二叉树不为空时，寻找根节点到叶子结点的路径，并判断和
        //这是对寻找叶子结点路径算法的扩展
        vector<int> ivec;//ivec用来保存路径
        FindPath(ivecvec, root, ivec, 0, expectNumber);
        return ivecvec;
    }
    //path用来保存路径, currSum表示当前路径的和
    void FindPath(vector<vector<int> > &ivecvec, TreeNode *root, vector<int> path, int currSum, const int expectNumber) {
        if (NULL != root) {//结点不为空，存入路径数组中
            path.push_back(root -> val);
            currSum += root->val;
            //如果结点的左、右子树均为空，说明这个节点为叶子结点
            //如果是叶子结点就要判断路径和是否等于期待值
            if ((NULL == root->left) && (NULL == root->right)) {
                if (currSum == expectNumber) {
                    ivecvec.push_back(path);
                }
                return;
            }
            //如果结点的左子树不为空
            if (root->left) {
                FindPath(ivecvec, root->left, path, currSum, expectNumber);
            }
            
            //如果结点的右子树不为空
            if (root->right) {
                FindPath(ivecvec, root->right, path, currSum, expectNumber);
            }
        }
        return;
    }
        
    };