给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ \
7 2 1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。
解析:虽然这个灰色的框很难看,但是手动调整空格又太麻烦,需要一个个的打
有谁知道怎么办吗。。。
本题的问题同之前的题目,即必须到达子节点,如果半途中sum就已经满足了也是不合法的。
迭代与递归两种解法。
原理都是一样的。
class Solution {
public:
//递归
bool hasPathSum(TreeNode* root, int sum) {
if(!root) return false;
if(!root->left&&!root->right&&root->val==sum) return true;
if(hasPathSum(root->left,sum-root->val)) return true;
return hasPathSum(root->right,sum-root->val);
}
//迭代
bool hasPathSum(TreeNode* root, int sum){
if(!root) return false;
stack<pair<TreeNode*,int>> sp;
sp.push(make_pair(root,sum));
while(!sp.empty()){
auto &item = sp.top();
auto node = item.first;
sum = item.second;
sp.pop();
if(!node->left&&!node->right&&node->val==sum) return true;
if(node->left) sp.push(make_pair(node->left,sum-node->val));
if(node->right) sp.push(make_pair(node->right,sum-node->val));
}
return false;
}
};
二次元憨批
发布了112 篇原创文章 · 获赞 0 · 访问量 331
私信
关注