给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ,判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。
叶子节点 是指没有子节点的节点。
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:true
解:
如果需要搜索整颗⼆叉树,那么递归函数就不要返回值,如果要搜索其中⼀条符合条件的路径,递归函数就需要返回值,因为遇到符合条件的路径了就要及时返回
在⼆叉树:我的左下⻆的值是多少? 中,因为要遍历树的所有路径,找出深度最深的叶⼦节点,所以递归函数不要
返回值。
⽽本题我们要找⼀条符合条件的路径,所以递归函数需要返回值,及时返回,那么返回类型是bool?
class Solution {
public:
bool hasPathSum(TreeNode* root, int targetSum) {
if(!root) return 0;
return traversal(root, targetSum - root -> val);
}
bool traversal(TreeNode* cur, int count) { //注意这一块代码
if (count == 0 && !cur -> left && !cur -> right) {
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;
}
};