给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum1) {
vector<vector<int>> ret;
if(root == NULL)
{
return ret;
}
TreeNode* p = root;
TreeNode* r;
int sum = 0;
bool flag;
stack<TreeNode*> s1;
stack<TreeNode*> s2;
do
{
while(p != NULL)
{
s1.push(p);
s2.push(p);
p = p->left;
} //找到最左下角的节点,这个节点没有左子树
r = NULL;
flag = true;
while(!s1.empty() && flag)
{
p = s1.top();
if(p->right == r)
{
if(p->left == NULL && p->right == NULL) //叶子节点
{
int numtmp = 0;
vector<int> mytmp;
for(int i = 0; i<s1.size(); i++)
{
TreeNode* tmp = s2.top();
numtmp = numtmp + (tmp->val);
mytmp.insert(mytmp.begin(),tmp->val);
s2.pop();
}
if(numtmp == sum1)
{
ret.push_back(mytmp);
}
}
s2 =s1;
p = s1.top();
r = p;
s1.pop();
s2.pop();
}
else
{
p = p->right;
flag = false;
}
}
}while(!s1.empty());
return ret;
}
};