Java 题解 | #牛群的最长距离#
牛群的最长距离
https://www.nowcoder.com/practice/82848c6aa1f74dd1b95d71f3c35c74d5
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类 * @return int整型 */ private int max_length = Integer.MIN_VALUE; public int diameterOfBinaryTree(TreeNode root) { if (root == null) { return 0; } dfs(root, 0); return max_length; } private int dfs(TreeNode root, int length) { if (root == null) { return 0; } int left = dfs(root.left, length + 1); int right = dfs(root.right, length + 1); max_length = Math.max(max_length, left + right); return Math.max(left, right) + 1; } }
该代码所使用的编程语言是Java。
通过这段代码,可以考察以下知识点:
- 二叉树的遍历和递归:代码中使用递归的方式遍历二叉树,分别在左子树和右子树中计算路径长度。
- 二叉树节点定义和构建:代码中使用了一个自定义的 TreeNode 类来表示二叉树的节点,并使用构造函数初始化节点对象。
- 逻辑判断和条件语句:代码中使用了逻辑判断语句 if 来进行条件判断和结果返回。
- 数值比较和取最大值:通过比较两个数值的大小,使用 Math.max 方法选取较大值更新最大直径长度。
- 全局变量的使用:通过定义全局变量 max_length,可以在递归过程中记录最大直径长度。
代码的文字解释大纲如下:
- 首先定义一个 TreeNode 类,表示二叉树的节点,包含一个整型的 val 属性以及左右子节点的引用。
- 在 Solution 类中定义一个方法 diameterOfBinaryTree,接收一个二叉树的根节点 root,返回值为整型,表示二叉树的直径。
- 全局变量 max_length 初始值设为整型的最小值,用来保存最大直径的长度。
- 在 diameterOfBinaryTree 方法中,判断根节点是否为空,如果为空则直接返回 0。
- 调用 dfs 方法递归地进行深度优先搜索,同时传入初始长度 0。最终结果保存在全局变量 max_length 中。
- 返回 max_length 的值作为二叉树的直径。
- 在 dfs 方法中,传入当前节点和当前路径长度。如果当前节点为空,返回 0。
- 分别递归地进入当前节点的左子树和右子树,并将路径长度加 1。
- 计算左子树和右子树的路径长度之和,并与全局变量 max_length 进行比较,取较大值更新 max_length。
- 返回左子树和右子树中的较大路径长度加上 1,表示当前节点的路径长度。
通过递归遍历二叉树,计算每个节点的左子树和右子树的路径长度,找到二叉树的最大直径。其中,直径定义为两个节点之间的最长路径,路径的长度是指路径上的边数。