题目介绍

leetcode.cn/problems/di…

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

示例

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

解题思路

1. 如何求最大直径：遍历所有的TreeNode节点，计算经过每个node节点的直径 d1,d2,d3,...dn，那么最大直径即为Math.max(d1,d2, ... dn)
2. 如何求经过node节点的直径呢？对于节点1来说，直径（3） = 左子树的深度（2） + 右子树的深度 （1）
3. 如何求一个节点的深度呢？这个比较简单，属于二叉树的基础

LC

    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        } 

        int left = maxDepth(root.left);
        int right = maxDepth(root.right);
        return Math.max(left, right) + 1;
    }

结论： 在计算二叉树的最大深度时，不断更新其最大直径的结果值。

解题代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    private int ans = 0;
    public int diameterOfBinaryTree(TreeNode root) {
        if (root == null) {
            return 0;
        }

        getHeight(root); 
        return ans;
    }

    private int getHeight(TreeNode root) {
        if (root == null) {
            return 0;
        }

        int l = getHeight(root.left);
        int r = getHeight(root.right);
        /** 
        更新直径的最大值
           4   对于节点4, l = 2 r = 1
          3  5  
         1  
        */  
        ans = Math.max(ans, l + r);
        return Math.max(l,r) + 1;  // 返回节点的深度
    }
}