1、题目描述
给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
2、算法分析
题目要求的是:二叉树里节点值之和等于targetSum的路径的数目。
符合条件的路径可能是:
① 路径包含root根节点
② 路径不包含根节点,并分别分布在除根节点之外的左右子树
具体看代码,注释很清晰。
3、代码实现
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
/**
使用targetSum - root.val == 0,说明这条路径上的结点是存在的
*/
class Solution {
public int pathSum(TreeNode root, int targetSum) {
if(root == null){
return 0;
}
/**
pathSumStartWithRoot(root,targetSum):因为前面判断过了root是否为空了,也就是算上根节点。
pathSum(root.left,targetSum) + pathSum(root.right,targetSum):需要重新判断除去 左右结点是否为空。
*/
int result = pathSumStartWithRoot(root,targetSum) + pathSum(root.left,targetSum) + pathSum(root.right,targetSum);
return result;
}
//
public int pathSumStartWithRoot(TreeNode root,int targetSum){
if(root == null){
return 0;
}
// 定义结果集,初始化为0,每次递归初始化result结果
int result = 0;
// 找到匹配的值的时候,result结果+1。这其中包含了一种情况。
if(targetSum - root.val == 0){
result++;
}
result += pathSumStartWithRoot(root.left,targetSum - root.val) + pathSumStartWithRoot(root.right,targetSum - root.val);
return result;
}
}