题目描述

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

样例

给定如下二叉树，以及目标和 sum = 22，
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

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

算法1

dfs

时间复杂度

时间复杂度分析：每个节点仅被遍历一次，且递归过程中维护 sum 的时间复杂度是 O(1)，所以总时间复杂度是 O(n)

C 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

bool dfs(struct TreeNode* root, int sum)
{
    sum -= root->val;
    if (sum == 0 && root->left == NULL && root->right == NULL) {
        return true;
    }


    if (root->left && dfs(root->left, sum)) { // if (root->left) return dfs(root->left, sum) {
        return true;
    }
    if (root->right && dfs(root->right, sum)) {
        return true;
    }
    return false;
}

bool hasPathSum(struct TreeNode* root, int sum){
    if (NULL == root) {
        return false;
    }
    return dfs(root, sum);
}