[菜鸟训练]437. 路径总和 III

题目描述：

给定一个二叉树，它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。
二叉树不超过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

解题思路：

使用双重递归，第一重递归用来遍历树上所有的节点，第二重递归用来进行搜索dfs操作。
我们就遍历所有的节点，然后让该节点为起始节点来进行搜索查询其是否存在一条路径，使得路径上的val和为targetSum。其中总的路径数用一个全局变量存储。详见代码。时间复杂度O(n²)

代码：

public class LC437 {

    //双重递归，第一重递归用来遍历树上所有的节点，第二重递归用来搜索dfs
    //时间复杂度O(n^2)
    int cnt;
    //将每个节点都作为起始点查看是否存在满足条件的路径
    public int pathSum(TreeNode root, int targetSum) {
        if (root == null){
            return 0;
        }
        search(root,targetSum);
        pathSum(root.left,targetSum);
        pathSum(root.right,targetSum);
        return cnt;
    }
    //以root为开始节点查找是否存在和为targetSum的路径
    private void search(TreeNode root, int targetSum) {
        if (root == null){
            return;
        }
        targetSum -= root.val;
        if (targetSum == 0){
            cnt++;
        }
        search(root.left,targetSum);
        search(root.right,targetSum);
    }
}