LeetCode437. 路径总和 III(双重递归)

https://leetcode-cn.com/problems/path-sum-iii/submissions/
给定一个二叉树,它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数。

示例:
root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8

      10
     /  \
    5   -3
   / \    \
  3   2   11
 / \   \
3  -2   1

返回 3。和等于 8 的路径有:

  1. 5 -> 3
  2. 5 -> 2 -> 1
  3. -3 -> 11

思路:双重递归

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

int find_path_another(struct TreeNode* root, int sum){
    if(NULL == root){
        return 0;
    }
    int res = 0;
    sum -= root->val;
    if(0 == sum){
        res++;
    }
    
    res += find_path_another(root->left, sum);
    res += find_path_another(root->right, sum);
    return res;
}

int find_path(struct TreeNode* root, int sum, int s){
    if(NULL == root){
        return 0;
    }
    
    int nn = 0;
    nn = find_path_another(root, s);
    nn += find_path(root->left, sum, s);
    nn += find_path(root->right, sum, s);
    return nn;
}

int pathSum(struct TreeNode* root, int sum) {
    return  find_path(root, sum, sum);
}

猜你喜欢

转载自blog.csdn.net/qq_34595352/article/details/88568275