题目来自于leetcode112 路径总和
给你二叉树的根节点 root
和一个表示目标和的整数 targetSum
。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum
。如果存在,返回 true
;否则,返回 false
。
叶子节点 是指没有子节点的节点。
bool helper(TreeNode* root, int sum, int targetSum) {
if (root == nullptr) {
return false;
}
sum += root->val;
if (sum == targetSum && root->left == nullptr && root->right == nullptr) {
return true;
}
return helper(root->left, sum, targetSum) || helper(root->right, sum, targetSum);
}
bool hasPathSum(TreeNode* root, int targetSum) {
return helper(root, 0, targetSum);
}
每轮递归都有自己的sum,像是一个存档,所以可以保证sum里总是由root延伸下来的。
最开始我用了&sum,导致一轮递归结束后root处的值丢失了,不好处理。
bool递归return false就是返回,return true就是继续
也可以用void写帮助函数,需要有额外的定义
void helper(TreeNode* root, int sum, int targetSum,bool& result) {
if (root == nullptr) {
return;
}
sum = sum + root->val;
if (sum == targetSum && root->left == nullptr && root->right == nullptr) result = true;
helper(root->left, sum, targetSum, result);
helper(root->right, sum, targetSum, result);
}
bool hasPathSum(TreeNode* root, int targetSum) {
int sum = 0;
bool result = false;
helper(root, sum, targetSum, result);
return result;
}