题解 | #牛群特殊路径的数量#
牛群特殊路径的数量
https://www.nowcoder.com/practice/1c0f95d8396a432db07945d8fe51a4f5?tpId=354&tqId=10595825&ru=/exam/oj&qru=/ta/interview-202-top/question-ranking&sourceUrl=%2Fexam%2Foj%3Fpage%3D1%26tab%3D%25E7%25AE%2597%25E6%25B3%2595%25E7%25AF%2587%26topicId%3D354
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整型
*/
public int pathSum (TreeNode root, int sum) {
// write code here
if (root == null) {
return 0;
}
int pathsFromRoot = countPathsFromNode(root, sum, 0);
int pathsFromLeft = pathSum(root.left,sum);
int pathsFromRight = pathSum(root.right, sum);
return pathsFromRoot + pathsFromLeft + pathsFromRight;
}
private static int countPathsFromNode(TreeNode node, int targetSum,
int currentSum) {
if (node == null) {
return 0;
}
currentSum += node.val;
int pathsCount = 0;
if (currentSum == targetSum) {
pathsCount++;
}
pathsCount += countPathsFromNode(node.left, targetSum, currentSum);
pathsCount += countPathsFromNode(node.right, targetSum, currentSum);
return pathsCount;
}
}
知识点:
- 基本的Java语法和概念。
- 二叉树的遍历。
- 递归算法的应用。
解题思路:
- 从根节点开始,递归地向左子树和右子树搜索,计算以当前节点为起点的路径数目。
- 对于每个节点,从当前节点出发,向下递归搜索路径,同时记录当前路径上的节点值之和。
- 如果当前路径上的节点值之和等于目标值,增加路径计数。
- 返回路径计数。
解释:
在这个实现中,我们首先从根节点开始递归地计算以当前节点为起点的路径数目。在递归过程中,我们记录当前路径上的节点值之和,如果等于目标值,则增加路径计数。然后,我们分别递归地计算左子树和右子树中满足条件的路径数目,并将它们累加起来。最终,我们可以得到总的路径数目。
安克创新 Anker公司福利 559人发布
