标签:leetcode
Given a binary tree and a sum, find all root-to-leaf paths where each path‘s sum equals the given sum.
For example:sum
= 22,
5
/ 4 8
/ / 11 13 4
/ \ / 7 2 5 1
return
[ [5,4,11,2], [5,8,4,5] ]
啊采用DFS方法,每次到达叶子节点返回,并且恢复原来的值!
代码如下:
class Solution {
public:
vector<vector<int>> pathSumVector;
int depthSum1;
vector<vector<int>> pathSum(TreeNode* root, int sum)
{
depthSum1=0;
if (root==NULL)
{
return pathSumVector;
}
vector<int> solution;
DFS_sum(root,sum,solution);
return pathSumVector;
}
void DFS_sum(TreeNode* root, int sum, vector<int> &solution)
{
if (root)
{
if (root->left==NULL&&root->right==NULL)
{
if (depthSum1+root->val==sum)
{
solution.push_back(root->val);
pathSumVector.push_back(solution);
solution.pop_back();
}
return;
}
depthSum1+=root->val;
solution.push_back(root->val);
DFS_sum(root->left,sum,solution);
DFS_sum(root->right,sum,solution);
depthSum1-=root->val;
solution.pop_back();
}
}
};版权声明:本文为博主原创文章,未经博主允许不得转载。
标签:leetcode
原文地址:http://blog.csdn.net/q286989429/article/details/47446185