剑指offer 面试题34 二叉树中和为某一值的路径

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

tips: 树的先序遍历，使用递归，用vector记录路径。

#include<iostream>
#include<vector>
#include<stack>
#include<algorithm>
#include<queue>
using namespace std;


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> > res;
		if(root==NULL) {
			return res;
		}
		vector<int> path;
		recursiveFindPath(root,path,res,expectNumber);
		return res;
    }
	void recursiveFindPath(TreeNode* root,vector<int> path,vector<vector<int>> & res,int expectNumber) {
		path.push_back(root->val);
		if(root->left==NULL&&root->right==NULL) {
			int pathSum=0;
			for (int i = 0; i < path.size(); i++)
			{
				pathSum+=path[i];
			}
			if(pathSum==expectNumber) {
				res.push_back(path);
			}
		}
		if(root->left!=NULL) {
			recursiveFindPath(root->left,path,res,expectNumber);
		}
		if(root->right!=NULL) {
			recursiveFindPath(root->right,path,res,expectNumber);
		}
	}
};

int main() {
	TreeNode* test=new TreeNode(10);
	test->left=new TreeNode(5);
	test->left->left=new TreeNode(4);
	test->left->right=new TreeNode(7);
	test->right=new TreeNode(12);
	Solution sol;
	vector<vector<int>> res=sol.FindPath(test,22);
	for (int i = 0; i < res.size(); i++)
	{
		for (int j = 0; j < res[i].size(); j++)
		{
			cout<<res[i][j]<<" ";
		}
		cout<<endl;
	}

	system("pause");
	return 0;
}