路经总和III
题目描述
想法
这种问题类似于前缀和问题,首先想到的解法是用HashMap来记录遍历到某个节点时整条路径上所以节点的值的和,但是这种方式无法区分所得到的满足条件的解中所有节点都在同一条路径上,(数组是由于路径的唯一性是可以利用这种方式来解决的,具体做法参见leetcode560和为K的数组),想到树的递归特性,考虑利用递归问题来解决这个问题。
题目中提出了三个要求:
- 路径不需要从根节点开始
- 路径不需要在叶子节点结束
- 路径方向只能向下,不能回头
所以给定一个根节点,所有的路径应该包括三部分:
- 以根节点作为头节点的路径
- 以根节点左孩子作为头节点的路径
- 以根节点右孩子作为头节点的路径
容易想到的是,当前节点为null时,上述三种路径都为0;
整个算法的流程是:
对于给定的根节点:
if(root == null) return 0;
else:
total = 路径中包含当前节点,以根节点作为头节点的路径数量
+ 路径中不包含当前节点,以根节点左孩子作为头节点的路径数量
+ 路径中不包含当前节点,以根节点右孩子作为头节点的路径数量
这里要说明,第二三种情况的语义是所得路径种不含当前的根节点
下面给为了体现递归,将根节点叫做当前节点
以当前节点作为头节点的路径:
要注意的是,在选定当前节点作为头节点时,它需要继续向下递归,所需要的和为sum - curr_root.val,这时不能使用上述算法进行计算,因为我们指定新的根为curr_root = curr_root.left或者curr_root = curr_root.right时,它计算的只是以当前节点作为头节点,且和为sum - curr_root.val的路径,否则,找到的路径可能是 不连续的
为此 我们需要另一个函数
public int pathSum(TreeNode root, int sum):计算的是上述三种路径的和
public int getCount(TreeNode curr,int sum):计算第一种路径的数量
到现在,算法框架已经比较清晰,可以用以下代码描述
class Solution {
// 方法入口
public int pathSum(TreeNode root, int sum) {
if(root == null) {
return 0;
}
// 后面两项为不包含当前的根节点且满足条件的路径数量,在这里,由于当前根节点没有添加到就中去 num 是不变的
return getCount(root,sum) + pathSum(root.left,sum) + pathSum(root.right,sum);
}
public int getCount(TreeNode curr,int sum) {
int root = 0, left = 0, right = 0;
if(curr == null) {
return 0;
}
sum = sum - curr.val;
root = sum == 0 ? 1:0;
// left 和 right 是深度优先搜索的过程,即我们已经将当前的根节点添加到了一个路径中,我们要继续深入搜索,看在路径中添加几个节点之后是否能满足条件。和上面方法中不一样的地方在于,这里的num是已经变化了的
left = getCount(curr.left,sum);
right = getCount(curr.right,sum);
return root + left + right;
}
}