import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @param sum int整型
* @return int整型
*/
int ans = 0;
public int pathSum (TreeNode root, int sum) {
// write code here
if (root == null) {
return 0;
}
preorder(root, sum);
return ans;
}
private void preorder(TreeNode root, int sum) {
if (root == null) {
return;
}
dfs(root, sum);
preorder(root.left, sum);
preorder(root.right, sum);
}
private void dfs(TreeNode root, int sum) {
if (root == null) {
return;
}
sum -= root.val;
if (sum == 0) {
ans++;
}
dfs(root.left, sum);
dfs(root.right, sum);
}
}
该代码使用的编程语言是Java。
这个题目考察的是二叉树的遍历和递归算法。
代码的详细文字解释如下:
- 首先定义了一个 TreeNode 结构体,表示二叉树的节点。它包含一个整型的值 val,以及左右子树的指针 left 和 right。
- 接下来是 Solution 类,其中定义了一个私有变量 ans,用于记录路径和等于给定值的路径数目。
- pathSum 方法是解决问题的入口函数。它接收一个二叉树的根节点和一个整数值作为参数,并返回路径和等于给定值的路径数目。在方法内部,首先判断根节点是否为空,如果为空,则直接返回0。然后调用 preorder 方法来遍历二叉树,同时传递给它根节点和目标值。最后,返回路径数目 ans。
- preorder 方法是一个前序遍历的递归函数,用于遍历二叉树的每个节点。它接收一个节点和目标值作为参数,首先判断节点是否为空,如果为空,则直接返回。然后调用 dfs 方法来计算以当前节点为起点的路径和等于目标值的路径数目。接着递归调用 preorder 方法遍历左子树和右子树。
- dfs 方法是一个深度优先搜索的递归函数,用于计算以当前节点为起点的路径和等于目标值的路径数目。它接收一个节点和目标值作为参数,首先判断节点是否为空,如果为空,则直接返回。然后将目标值减去当前节点的值。如果目标值等于0,则说明找到了一条路径,将路径数目 ans 增加1。最后,递归调用 dfs 方法计算左子树和右子树中路径和等于目标值的路径数目。
通过递归遍历二叉树的每个节点,计算出以每个节点为起点的路径和等于目标值的路径数目,并累加到 ans 变量中,最终得到路径和等于给定值的路径数目。

京公网安备 11010502036488号