题目(leecode T112):
给你二叉树的根节点 root
和一个表示目标和的整数 targetSum
。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum
。如果存在,返回 true
;否则,返回 false
。
叶子节点 是指没有子节点的节点。
方法:本题我们需要找二叉树中所有路径中和符合给定条件的路径,并返回bool值。使用迭代法需要考虑三要素:
1:参数和返回值:参数包括树节点和需要求的target值,返回是bool类型,判断是否成功找到了路径和符合给定条件。
2:确定终止条件:终止条件有两个,当找到符合要求的路径时返回,具体是当找到了叶子节点并且和等于给定的target值,返回true;还有一个是到了叶子节点但路径和不满足要求,返回false
3:确定单层处理逻辑:因为递归函数是有返回值的,如果递归函数返回true,说明找到了合适的路径,应该立刻返回。
题解:
class Solution {
private:
bool traversal(TreeNode* cur, int count){
if(!cur->left && !cur->right && count ==0) return true; //找到符合要求的路径
if(!cur->left && !cur->right) return false; //找到叶子节点但和不满足
if(cur->left){
count -= cur->left->val;
if(traversal(cur->left, count)) return true;
count += cur->left->val; //不满足条件就回溯
}
if(cur->right){
count -= cur->right->val;
if(traversal(cur->right, count)) return true;
count += cur->right->val;
}
return false;
}
public:
bool hasPathSum(TreeNode* root, int targetSum) {
if(root == NULL) return false;
return traversal(root, targetSum - root->val);
}
};