Path Sum II
最后更新于:2022-04-01 22:56:14
## 一.题目描述
Given a binary tree and a sum, find all root-to-leaf paths where each path’s sum equals the given sum.
For example:
Given the below binary tree and `sum = 22`,
~~~
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
~~~
return
~~~
[
[5,4,11,2],
[5,8,4,5]
]
~~~
## 二.题目分析
这道题与Path Sum类似,不同的是这道题要求找到所有和等于给定值的路径,必须遍历完所有的路径。在找到满足的路径之后,不能直接返回,而是将其添加到一个`vector>`中。在查找的过程中,每经过一个结点,先使用一个`vector`将该路径中的所有结点记录下来。需要注意的是,在进入每一个结点的时候,先将该结点的值`push`到`vector`中,在退出时将该结点的值`pop`出来,这样就可以避免有时会忘记`pop`结点的值的情况。
该方法的时间复杂度为O(n),空间复杂度为O(logn)。
## 三.示例代码
~~~
struct TreeNode
{
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution
{
public:
vector> pathSum(TreeNode* root, int sum)
{
vector > Result;
vector TmpResult;
pathSum(root, sum, Result, TmpResult);
return Result;
}
private:
void pathSum(TreeNode* root, int sum, vector > &Result, vector &TmpResult)
{
if (!root)
return;
if (!root->left && !root->right && root->val == sum)
{
TmpResult.push_back(sum);
Result.push_back(TmpResult);
// pop the leaf node
TmpResult.pop_back();
return;
}
int SumChild = sum - root->val;
TmpResult.push_back(root->val);
pathSum(root->left, SumChild, Result, TmpResult);
pathSum(root->right, SumChild, Result, TmpResult);
TmpResult.pop_back();
}
};
~~~
## 四.小结
与Path Sum一样,用DFS解决。
';