437. 路径总和 III

题目

给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出：3
解释：和等于 8 的路径有 3 条，如图所示。

思路 （宫水三叶）

用dfs1遍历每个节点，用dfs2统计以该节点为起点，所有可能的路径，中间统计为target的数量。

• 时间复杂度：O(n^2)
• 空间复杂度：忽略递归带来的额外空间开销，复杂度为 O(1)

盲点

用例里有超过int的max的值，所以用long

代码

package binaryTree;

class o437 {
    int res = 0;
    int targetSum;

    public int pathSum(TreeNode root, int targetSum) {
        this.targetSum = targetSum;
        dfs1(root);
        return res;
    }

    private void dfs1(TreeNode root) {
        if (root == null) {
            return;
        }
        dfs2(root, root.val);
        dfs1(root.left);
        dfs1(root.right);
    }

    private void dfs2(TreeNode root, int val) {
        if (val == targetSum) {
            res++;
        }
        if (root.left != null) {
            dfs2(root.left, val + root.left.val);
        }
        if (root.right != null) {
            dfs2(root.right, val + root.right.val);
        }
    }

    public static void main(String[] args) {
        TreeNode n3 = new TreeNode(3);
        TreeNode nn2 = new TreeNode(-2);
        TreeNode n1 = new TreeNode(1);
        TreeNode n32 = new TreeNode(3, n3, nn2);
        TreeNode n2 = new TreeNode(2, null, n1);
        TreeNode n11 = new TreeNode(11);
        TreeNode nn3 = new TreeNode(-3, null, n11);
        TreeNode n5 = new TreeNode(5, n32, n2);
        TreeNode root = new TreeNode(10, n5, nn3);
        o437 main = new o437();
        int targetSum = 8;
        System.out.println(main.pathSum(root, targetSum));
    }
}