给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22
,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
----------------------------------------------------------------------------------------------------------------------------------------------------------------------------
这道题,学到了,真的是学到了,直接把路径总和lll 也AC了,hahaha.
一开始,想不通这个路径该怎么保存,可以由一个 vector<int> t 和一个 vector<vector<int>> s 来实现路径的保存。
思路:递归到叶子节点,并判断这是是否和之前的和为sum,如果等于,直接s.push_back(t); 实现这条路的保存。
再将最后进入的弹出:t.pop_back() 最后路径的条数为s.size(),并且路径都在容器s中。
AC:
class Solution {
public:
vector<vector<int>> s;
vector<int> t;
void fun(TreeNode* root,int sum) //这里的函数不需要返回,因为结果都在s中,s是全局变量。
{
if (root==NULL)
return ;
t.push_back(root->val);
if (root->val==sum&&root->left==NULL&&root->right==NULL)
{
s.push_back(t);
}
fun(root->left,sum-root->val);
fun(root->right,sum-root->val);
t.pop_back(); //将最近push()进的叶子节点弹出,实现换路。
}
vector<vector<int>> pathSum(TreeNode* root, int sum) {
fun(root,sum);
return s;
}
};
这道题真挺好的,学到了很多,666666666送给我自己,hahahhaaha(来自菜鸟的欣喜).
容器真是个好东西,得好好看看。