leetcode437. 路径总和 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

解题

• 递归
• 这里使用了两个递归方法, 一个递归每个节点, 以每个节点为起始点查找sum
• 一个是以当前节点为起始点, 所有和为sum的路径, 代码如下

class Solution {
    private int result = 0;
    public int pathSum(TreeNode root, int sum) {
        findSum(root, sum);
        return result;
    }

    private void findSum(TreeNode root, int sum){
        if (root == null) {
            return;
        }
        findPath(root, sum);
        findSum(root.left, sum);
        findSum(root.right, sum);

    }

    // 查找以root为根节点的所有和为sum的路径
    private void findPath(TreeNode root, int sum) {
        if (root == null) {
            return;
        }

        if (root.val == sum) {
            result++;
        }

        findPath(root.left, sum - root.val);
        findPath(root.right, sum - root.val);
    }

}