LeetCode437-路径总和III

发表于 | 分类于

题目链接

英文链接：https://leetcode.com/problems/path-sum-iii/

中文链接：https://leetcode-cn.com/problems/path-sum-iii/

题目详述

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

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

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

二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。

示例：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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

题目详解

本题与 LeetCode112-路径总和 和 LeetCode113-路径总和II 的主要区别在于：前面两题都要求路径是从根节点到叶子节点，而本题路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

  • 为了方便，用方法 pathSumFrom 记录从某个结点开始的路径总数。
  • 从根结点开始，调用 pathSumFrom 得到从根结点开始的路径总数；再分别递归得到左子树和右子树的路径总数，三者的和就是最终的结果。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
public class LeetCode_00437 {

    public int pathSum(TreeNode root, int sum) {
        if (root == null) {
            return 0;
        }
        return pathSumFrom(root, sum) + pathSum(root.left, sum) + pathSum(root.right, sum);
    }

    private int pathSumFrom(TreeNode root, int sum) {
        if (root == null) {
            return 0;
        }
        sum -= root.val;
        return (sum == 0 ? 1 : 0) + pathSumFrom(root.left, sum) + pathSumFrom(root.right, sum);
    }
}