Path Sum:
Given a binary tree and a sum, determine if the tree has a root-to-leaf path
such that adding up all the values along the path equals the given sum.
For example:
Given the below binary tree
and sum = 22
,
5
/ 4 8
/ / 11 13 4
/ \ 7 2 1
return true, as there exist a root-to-leaf
path 5->4->11->2
which sum is 22.
解题分析:
二叉树首先应该想到递归
对于root = 5, 寻找sum=22的路径 这个问题可以分为 root=4 寻找sum=22-5=17的路径 和 root=8
寻找sum=22-5=17的路径 这两部分
也就是说,我们可以递归寻找左右子树路径和,我们只需要更改和值即可
终止条件:如果root为空,立即返回false, 如果root为叶子节点,如果其值为最后剩余的和值,那么就存在总的路径和值等于sum
class Solution {
public:
bool hasPathSum(TreeNode *root, int sum) {
if (root == nullptr) return false;
if (root->left == NULL && root->right == NULL) return sum == root->val;
return hasPathSum(root->left, sum - root->val) || hasPathSum(root->right, sum - root->val);
}
};
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]
]
解题分析:
现在需要求出满足条件的路径,并且是所有路径,所以左子树求到了满意结果,不能return,要接着求右子树
我们在上一题中,最后叶子结点如果满足 val == sum 那么这条路径就符合要求,此时我们将结果添加到result中
但是如果保存该路径上的所有结点值呢? 我们是 每遇到一个非空结点都添加到一个vector中(其实这个vector就是一个栈)
如果栈顶(vector最末元素)的左右子树都不存在满足条件的路径,那么就出栈,也就是 从其父节点开始继续判断
这就是类似于 二叉树的先序遍历,每遇到一个结点就入栈,随后如果不满足条件就出栈达到其父节点
先序遍历对应 根左右,也就是说最后的结果就是 从根节点出发的路径,即路径方向也吻合,而不需要逆向
class Solution {
public:
vector<vector<int> > pathSum(TreeNode *root, int sum) {
vector<vector<int>> result;
vector<int> tmp;
BintreePathSum(root, sum, tmp, result);
return result;
}void BintreePathSum(TreeNode* root, int gap, vector<int>& tmp, vector<vector<int>>& result) {
if (root == NULL) return;
tmp.push_back(root->val);if (root->left == NULL && root->right == NULL) {
if (root->val == gap) {
result.push_back(tmp);
}
}
BintreePathSum(root->left, gap - root->val, tmp, result);
BintreePathSum(root->right, gap - root->val, tmp, result);tmp.pop_back();
}
};
Leetcode:Path Sum 二叉树路径和,布布扣,bubuko.com