题目来源:牛客网
题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
解析:
经典DFS算法,深度优先算法
class Solution {
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
vector<vector<int>>ret;
vector<int>path;
dfs(ret,root,path,expectNumber,0,0);
return ret;
}
void dfs(vector<vector<int>>&ret,TreeNode* root,vector<int>&path,int expectNumber,int len,int sum)
{
if(root == nullptr)
return;
if(len == path.size())
path.push_back(root->val);
else
path[len] = root->val;
if(sum + root->val == expectNumber && root->left == nullptr && root->right == nullptr)
{
vector<int>arr(len+1);
for(int i=0;i<len+1;i++)
{
arr[i] = path[i];
}
ret.push_back(arr);
}
dfs(ret,root->left,path,expectNumber,len+1,sum + root->val);
dfs(ret,root->right,path,expectNumber,len+1,sum + root->val);
}
};
经典的深度优先代码应该是这样的:
class Solution {
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
vector<vector<int>> ret;
vector<int> path;
dfs(root,ret,path,expectNumber,0);
return ret;
}
void dfs(TreeNode*root,vector<vector<int>>&ret,vector<int>&path,int n,int sum)
{
if(root==nullptr)
return;
path.push_back(root->val); //先把这个节点放进去
if(sum + root->val == n && root->left==nullptr && root->right==nullptr) //满足即可输出
{
ret.push_back(path);
//return ;
}
if(root->left!=nullptr)dfs(root->left,ret,path,n,sum+root->val); //不满足找左边的,之后再找右边的
if(root->right!=nullptr)dfs(root->right,ret,path,n,sum+root->val);
path.pop_back(); //找完左右两边的,都不满足,则把当前的这个pop掉
}
};