246. 二叉树的路径和 II(C++回溯法)
初学回溯,费了好半天时间,加油!!!!!!!!
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
public:
/*
* @param root: the root of binary tree
* @param target: An integer
* @return: all valid paths
*/
vector<vector<int>>res;//全局变量,保存结果
vector<TreeNode*>pre;//全局变量,保存所有节点
//先序遍历树,获取所有节点
void preOrder(TreeNode*tree)
{
if(tree==NULL)
{
return;
}
else
{
pre.push_back(tree);
preOrder(tree->left);
preOrder(tree->right);
}
}
//回溯法查找所有路径
void findPath(TreeNode*root, int &target, vector<int> &tmp)
{
if (root == NULL)//递归出口
return;
target -= root->val;
tmp.push_back(root->val);//把根节点尝试作为路径的一个环节
if(target == 0)//分两种情况
{
//若某条路径已经到达叶子节点
if(root->left==NULL&&root->right==NULL)
{
res.push_back(tmp);
tmp.pop_back();//回溯
target += root->val;//回溯
return ;//因为已经到达叶子节点,返回
}
else //该路径还没到达叶子节点
{
res.push_back(tmp);
//继续收索看还有没有更长的//如:1 1 1满足要求 1 1 1 1 -1 也满足要求
}
}
findPath(root->left, target, tmp);//递归查找左子树满足要求的所有路径
findPath(root->right, target, tmp);//递归查找右子树所有满足要求的路劲
tmp.pop_back();//回溯
target += root->val;//回溯
}
vector<vector<int>> binaryTreePathSum2(TreeNode * root, int target) {
// write your code here
preOrder(root);//获取所有节点
for(int i=0;i<pre.size();i++)
{
vector<int>tmp;//保存每一条路径
findPath(pre[i], target, tmp); //以每个节点为起点,查找所有满足要求的路径
}
return res;
}
};