面试题 04.12. 求和路径

给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

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

              5
             / 
            4   8
           /   / 
          11  13  4
         /      / 
        7    2  5   1

返回:

3
解释：和为 22 的路径有：[5,4,11,2], [5,8,4,5], [4,11,7]

提示：

• 节点总数 <= 10000

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* root,int sum,int &curSum,int &pathNum){
        if(!root)return;
        curSum+=root->val;
        if(curSum==sum)pathNum++;
        if(root->left)dfs(root->left,sum,curSum,pathNum);
        if(root->right)dfs(root->right,sum,curSum,pathNum);
        curSum-=root->val;
    }
    int pathSum(TreeNode* root, int sum) {
        if(!root)return 0;
        int curSum=0,pathNum=0;
        dfs(root,sum,curSum,pathNum);
        pathNum+=pathSum(root->left,sum);
        pathNum+=pathSum(root->right,sum);
        return pathNum;
    }
};