一 DFS 递归
思路 :
树的题目一般采用递归方法
分别判断 root 受否满足条件 递归左子树 ,递归右子树 判断
如何考虑呢 ?
先看作整体 ,找到满足条件时返回出口,这个时候可以 考虑只有三个节点情况,分别写出判断条件.然后递归左右子树.
1 判断root 节点是否为空 空return false
2 非空 判断是否满足条件
条件为叶子节点(左右节点都为空) 且sum==val return true
3 递归判断左右子树是否满足条件
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
if(!root) return false;
if(root->val==sum&&root->left==NULL&&root->right==NULL) return true;
return hasPathSum(root->left,sum-root->val)||hasPathSum(root->right,sum-root->val);
}
};
二 回溯法
参考 添加链接描述
三 栈迭代方法
将遍历的每个节点都放入到栈中 , 同时更新val值,
判断当为叶子节点时是否满足条件.
若非叶子节点 继续左右子树
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
if(!root) return false;
stack<TreeNode*> stk;
stk.push(root);
while(!stk.empty())
{
auto p=stk.top();
stk.pop();
if(p->val==sum&&!p->left&&!p->right){
return true;
}
if(p->left){
p->left->val+=p->val;
stk.push(p->left);
}
if(p->right){
p->right->val+=p->val;
stk.push(p->right);
}
}
return false;
}
};