题目:
两重递归思路:
两个递归,一个是遍历出发点的递归,另一个是递归计算从当前节点出发得到sum的路径数。
递归出发点:
如果当前节点为null,则返回0.如果当前节点非null节点,则计算当前节点是否为所求解并且以它的左子节点和右子节点为root递归该方法。
public int pathSum(TreeNode root, int sum) {
if(root==null){
return 0;
}else{
return dfs(root, sum)+pathSum(root.left, sum)+pathSum(root.right, sum);
}
}
递归计算从当前节点出发得到sum的路径数:
这个节点可以返回一个值,从当前节点出发有多少条可行的路径,这个值为0或者1或者2。它的思路是把sum减去当前节点的值,然后递归,若最后的节点刚好等于sum,那么就计数加一,最终会累加返回,其中遇到null会返回0。
public int dfs(TreeNode root,int sum){
int res=0;
if(root==null){
return 0;
}else if(root.val==sum){
res++;
}
res+=dfs(root.left, sum-root.val);
res+=dfs(root.right, sum-root.val);
return res;
}
巧妙的地方:
这段代码把所有可能的结果返回了。子系统的返回值一层层地累加上来,这里是左子树的路径和加上右子树的路径和再加上当前节点出发得到的路径数作为返回。多级的return连成一条通路就是这段代码最有趣的地方。
return dfs(root, sum)+pathSum(root.left, sum)+pathSum(root.right, sum);