leetcode543二叉树的直径

leetcode543 二叉树的直径

题目详情

定义：二叉树的直径

给定一个二叉树，二叉树的直径长度为任意两个节点路径长度的最大值。

定义：二叉树的深度

二叉树的深度为：从根节点到达叶子节点的最大路径长度，路径长度是指从根节点到达叶子节点的经过的节点的个数。

题目解析

根据题目我们可以发现求解二叉树的直径和求解二叉树的深度有相似之处，二叉树的深度为左子树的最大深度 L 加上右子树的最大深度 R 在加上1即（L + R + 1）。那么求解二叉树的直径可以想象成把任意节点当作根节点时最大的左右子树二叉树深度的和。由于题目中所述最大路径是指节点之间的边的数量，因此最终的结果应该减去1
为了能够记录最大深度，我们维持一个静态变量用于记录最大的深度。

public class Solution {
    public static int ans;
    public int diameterOfBinaryTree(TreeNode root){
        ans = 1;
        depth(root);
        return ans - 1;
    }
    private int depth(TreeNode root){
        if(root == null)
            return 0;
        int L = depth(root.left);
        int R = depth(root.right);
        ans = Math.max(ans, L + R + 1); //用于维持最大直径
        return Math.max(L, R) + 1; // 返回为以 root 为根节点时二叉树的最大深度
    }

    class TreeNode{
        int val ;
        TreeNode left;
        TreeNode right;
        TreeNode(int val){
            this.val = val;
        }
    }
}