（LC）543. 二叉树的直径

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

示例 :
给定二叉树

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

两种解法： 该问题的本质是求某个结点左右子树最大深度的和

• 解法1：暴力解法，遍历二叉树时前序位置无法获取子树的信息，所以只能调用maxDepth()去计算子树的最大深度，最坏时间复杂度为O(2^n)

/**
 * 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 {
    int maxDiameter = 0;
    public int diameterOfBinaryTree(TreeNode root) {
        traverse(root);
        return maxDiameter;
    }
    // 遍历二叉树
    public void traverse(TreeNode root) {
        if (root == null) {
            return;
        }
        // 对每个结点计算直径
        int maxLeft = maxDepth(root.left);
        int maxRight = maxDepth(root.right);
        int res = maxLeft + maxRight;
        // 更新全局最大直径
        maxDiameter = Math.max(maxDiameter, res);
        traverse(root.left);
        traverse(root.right);
    }
    // 计算二叉树的最大深度
    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int maxLeft = maxDepth(root.left);
        int maxRight = maxDepth(root.right);
        return Math.max(maxLeft, maxRight) + 1;
    }

}

• 解法二：在后序位置计算直径，因为在后序位置是知道子树的信息的，时间复杂度为O(n)

/**
 * 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 {
    int maxDiameter = 0;
    public int diameterOfBinaryTree(TreeNode root) {
        maxDepth(root);
        return maxDiameter;
    }
    // 计算二叉树的最大深度
    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int maxLeft = maxDepth(root.left);
        int maxRight = maxDepth(root.right);
        // 后序位置计算直径，提高效率
        maxDiameter = Math.max(maxDiameter, maxLeft + maxRight);
        return Math.max(maxLeft, maxRight) + 1;
    }

}