leetcode刷题（24）——437.路经总和III

一、题目

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

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

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

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

二、思路

二叉树的相关问题都可以考虑用递归解决。

递归思维的关键：（1）找到最简单的子问题，也就是递归结束条件；（2）其它问题不考虑内在细节，只考虑整体逻辑。

首先，递归结束条件：

if(root == null)
	return null;

然后，分析怎样构建递归结构。题目要求路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点），所以我们可以从根节点开始，计算：

• 以当前节点开头的路径数；
• 以当前节点的左孩子开头的路径数；
• 以当前节点的右孩子开头的路径数。

三部分之和即为总路径数。

最后，如何求得以当前节点开头的路径数？每到一个节点就判断 sum-root.val 是否为 0，如果为 0 的话，则该路径满足题目的条件。

三、代码

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

    // 计算以当前节点开头的满足和为sum的路径数
    public int countPath(TreeNode root, int sum){
        if(root == null)
            return 0;
        sum = sum - root.val;
        int t = sum == 0 ? 1 : 0;
        return t + countPath(root.left, sum) + countPath(root.right, sum);
    }
}