题目描述

给定一个二叉树和一个目标和，判断该树中是否存在根节点到叶子节点的路径，这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。

样例

给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

算法1

(暴力枚举) $O(n)$

遍历所有叶子节点路径，如果和为目标值，返回true。

时间复杂度

遍历所有节点，时间复杂度$O(n)$。

参考文献

C++ 代码

class Solution {
public:
    bool hasPathSum(TreeNode* root, int sum) {
        return dfs(root, sum, 0);
    }

    bool dfs(TreeNode *root, int target, int cur_sum){
        if (!root) return false;

        cur_sum += root->val;
        if (!root->left && !root->right){
            return cur_sum == target;
        }

        if (dfs(root->left, target, cur_sum)) return true;
        if (dfs(root->right, target, cur_sum)) return true;

        return false;
    }
};

更简洁的写法

class Solution {
public:
    bool hasPathSum(TreeNode* root, int sum) {
        if (!root) return false;
        if (!root->left && !root->right) 
            return sum == root->val;
        if (root->left && hasPathSum(root->left, sum - root->val)) 
            return true;
        if (root->right && hasPathSum(root->right, sum - root->val)) 
            return true;

        return false;
    }
};