本题我刚开始的想法是使用数组储存路径上的节点,遍历到叶子结点时再求和并与目标值进行比较,但这种做法使代码更加复杂,不仅要引入数组操作,还需要多个for循环用来求和。因此,本题采用递减法更加合适,没遍历一个路径上的节点,便将目标值减去节点的值,当遍历到叶子结点时判断目标值是否为0。
本题需要注意的几个点:
(1)递归函数是有返回值的,如果递归函数返回true,说明找到了合适的路径,应该立刻返回
(2)每次向下遍历应包含回溯
(3)头次遍历的目标值应减去根节点的值
代码如下:
class Solution {
private:
bool traversal(TreeNode* cur, int count) {
if (!cur->left && !cur->right && count == 0) return true; // 遇到叶子节点,并且计数为0
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 sum) {
if (root == NULL) return false;
return traversal(root, sum - root->val);
}
};