题解 | #牛群的最长距离# java

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整型
     */
        // write code here
        public int diameterOfBinaryTree(TreeNode root) {
            int[] diameter = new int[1]; // 直径长度
            depth(root, diameter);
            return diameter[0];
        }

        /**
         * 计算二叉树的深度，并更新直径的值
         *
         * @param node 当前遍历到的节点
         * @param diameter 直径数组，存储当前最大直径的值
         * @return 二叉树的深度
         */
        private int depth(TreeNode node, int[] diameter) {
            if (node == null) {
                return 0;
            }

            int left = depth(node.left, diameter); // 左子树的深度
            int right = depth(node.right, diameter); // 右子树的深度

            diameter[0] = Math.max(diameter[0], left + right); // 更新直径的值

            return Math.max(left, right) + 1; // 返回当前子树的深度
        }
    }

该代码使用的编程语言是Java。

该题考察的知识点是二叉树的深度和直径。

• diameterOfBinaryTree 方法用于计算二叉树的直径，即任意两个节点之间最长路径的长度。调用了 depth 方法来计算每个节点的深度，并同时更新直径的值。
• depth 方法是一个递归函数，用于计算二叉树中某个节点的深度。它接收一个节点和直径的引用作为参数，并返回当前子树的深度。在方法内部，首先判断节点是否为空，如果为空，则返回0。然后通过递归调用 depth 方法计算左子树和右子树的深度。使用 max 函数来更新直径的值，即取当前直径的最大值。最后，返回当前子树的深度，即左子树和右子树深度的较大值加上1。

这样，通过递归遍历二叉树的每个节点，计算出每个节点的深度，并不断更新直径的值，最终得到二叉树的直径。