题目大意
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
解题思路
首先判断当前节点是否是叶子结点:
(1)若是叶子结点,则判断从根节点到当前叶子结点的累加和是否等于目标和即可;
(2)不是叶子结点,则记录根节点到当前节点的累加和,然后递归地判断左右子树上是否存在一个路径,使得路径上的和加上当前累加和扥估目标和;
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
if (root == nullptr)
return false;
return judge(root, 0, sum);
}
private:
bool judge(TreeNode * root, int current, int sum)
{
if (root == nullptr)
return false;
if (root->left == nullptr && root->right == nullptr)
{
if (root->val + current == sum)
return true;
else
return false;
}
return judge(root->left, current + root->val, sum) || judge(root->right, current + root->val, sum);
}
};