给定一棵二叉树,其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法,打印节点数值总和等于某个给定值的所有路径的数量。注意,路径不一定非得从二叉树的根节点或叶节点开始或结束,但是其方向必须向下(只能从父节点指向子节点方向)。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/paths-with-sum-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
直接暴力解法,对root和树中的每一个子树遍历满足的路径数(由于不需要root为起点,所以所有节点对应的子树都要试试)
class Solution {
//h函数:统计以根节点为起点,所有满足条件的路径数
public int h(TreeNode root, int sum){
//null节点不存在,返回0
if(root==null){
return 0;
}else if(root.val==sum){
//如果根节点val等于sum,计数+1并且需要统计左右两边和为0的路径数
return 1+h(root.right,0)+h(root.left,0);
}else{
//否则,继续往下走
return h(root.right,sum-root.val)+h(root.left,sum-root.val);
}
}
public int pathSum(TreeNode root, int sum) {
if(root==null){
return 0;
}
//需要统计此节点出发路径数和左右子树所有元素出发的路径数
return h(root,sum)+pathSum(root.right,sum)+pathSum(root.left,sum);
}
}