冲击大厂算法面试=＞二叉树专题【二叉树的直径】

什么是二叉树的直径？

二叉树的直径是指：树中任意两个节点之间的最大距离。

求二叉树的直径

给你一棵二叉树的根节点，返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。
两节点之间路径的 长度 由它们之间边数表示。

示例 1：

输入：root = [1,2,3,4,5]
输出：3
解释：3 ，取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。

上代码

class Solution {
    // 最大直径就是两个节点的最大深度相加最大
    public int diameterOfBinaryTree(TreeNode root) {
        if (root == null) {
            return 0;
        }
        // 递归计算左子树的直径
        int leftDiameter = diameterOfBinaryTree(root.left);
        // 递归计算右子树的直径
        int rightDiameter = diameterOfBinaryTree(root.right);

        int left = maxDeep(root.left);

        int right = maxDeep(root.right);

        int maxDiameter = Math.max(leftDiameter, rightDiameter);

        return Math.max(maxDiameter, left + right);
    }

    // 求root的最大深度
    public int maxDeep(TreeNode root) {
        if (root == null) {
            return 0;
        }

        int left = maxDeep(root.left);

        int right = maxDeep(root.right);

        return Math.max(left, right) + 1;
    }
}

解释呀

方法diameterOfBinaryTree ：求二叉树的root的直径。
方法maxDeep：求二叉树的root的深度，用于辅助直径的计算。

假设有以下二叉树：
      1
     / \
    2   3
   / \
  4   5

深度计算:
  对于节点 4 和 5，它们的深度都是 1。
  对于节点 2，它的左子树深度是 1（来自节点 4），右子树深度是 1（来自节点 5），所以节点 2 的深度是 2（最大深度加 1）。
  对于节点 3，它是叶子节点，深度是 1。
  对于根节点 1，它的左子树深度是 2（来自节点 2），右子树深度是 1（来自节点 3），所以根节点的深度是 3（最大深度加 1）。

直径计算:
  节点 2 作为路径中间节点，左子树深度 1 + 右子树深度 1 = 2。
  节点 1 作为路径中间节点，左子树深度 2 + 右子树深度 1 = 3。
  左子树的直径是 2，右子树的直径是 1，所以树的直径是 3。

最终，记录的最大直径是 3，即节点 4 到节点 5 经过节点 2，再经过节点 1 的路径长度。

为什么？ Math.max(maxDiameter, left + right);
答：用于确保返回的直径是当前子树直径和之前计算的最大直径中的较大值。left + right 是当前节点作为路径中间点的直径，而 maxDiameter 是之前计算过的最大直径。这样可以确保无论是通过当前节点的路径还是子树的路径，最终得到的都是树的最大直径。

更简答理解

就是在求二叉树的深度这里维护一个最大值最后返回。

class Solution {
    int ans = 0;

    public int getMaxWidth(TreeNode root){
        if(root == null) return 0;
        //递归求左右子树最大高度
        int left = getMaxWidth(root.left);
        int right = getMaxWidth(root.right);

        //求递归过程中可能的最大直径
        ans = Math.max(ans, left + right);

        //你求你的最大值，管我当前节点的递归啥事，别拦我，我要返回子树的最大高度给上一层用
        return Math.max(left, right) + 1;
    }
    public int diameterOfBinaryTree(TreeNode root) {
        getMaxWidth(root);
        return ans;
    }
}