双重递归

给定一个二叉树,它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。

二叉树不超过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

思路:双重递归,这种题目需要从每个节点开始进行类似的计算,所以第一个递归用来遍历这些节点,这二个递归用来处理这些节点,进行深度优先搜索。

class Solution {
public:
    int count=0;
    int pathSum(TreeNode* root, int sum) {
        if(root==NULL) return 0;
        dfs(root,sum);
        pathSum(root->left,sum);
        pathSum(root->right,sum);
        return count;
    }
    void dfs(TreeNode* node, int sum)
    {
        if(node==NULL) return;
        if(sum-node->val==0) count++;
        dfs(node->left,sum-node->val);
        dfs(node->right,sum-node->val);
    }
};

 

posted @ 2020-07-16 18:11  江雨牧  阅读(655)  评论(0)    收藏  举报