题解 | JZ82 二叉树中和为某一值的路径(一)
这题要考虑到给的树为空树的情况,所以不能这么写递归出口:
if (!root) {
if (sum == 0) return true;
return false;
}
因为这样的话,会把树为空的情况给返回true,但答案应该为false。
所以只能判断到叶子节点这一层。
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
class Solution {
public:
/**
*
* @param root TreeNode类
* @param sum int整型
* @return bool布尔型
*/
bool hasPathSum(TreeNode* root, int sum) {
// write code here
if (!root) return false;
if (!root->left && !root->right) {
if (sum == root->val) return true;
return false;
}
return hasPathSum(root->left, sum-root->val) || hasPathSum(root->right, sum-root->val);
}
};
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
#
# @param root TreeNode类
# @param sum int整型
# @return bool布尔型
#
class Solution:
def hasPathSum(self , root: TreeNode, sum: int) -> bool:
# write code here
if root is None:
return False
if root.left is None and root.right is None:
if root.val == sum:
return True
return False
return self.hasPathSum(root.left, sum-root.val) or self.hasPathSum(root.right, sum-root.val)