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

题目描述

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

思路

递归根到叶子节点的所有路径，然后判断和是否为输入的值，是的话就将路径存储起来，注意路径必须是从根到叶子节点的路径。

代码

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