LeetCode - 543. 二叉树的直径

题目

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

思路

经过某个节点的路径长度中的最大值，就是它的左子树的高度 + 右子树的高度。
遍历所有的节点，取最大值，就是这个树的直径长度。

复杂度分析

假设树有 n 个节点。

• 时间复杂度O(n)，遍历整棵树。
• 空间复杂度O(n)，空间复杂度与整棵树的节点数相关。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private int ans;

    public int diameterOfBinaryTree(TreeNode root) {
        dfs(root);
        return ans;
    }

    private void dfs(TreeNode root) {
        if (root == null){
            return;
        }
        int diameter = depth(root.left) + depth(root.right);
        if (diameter > ans) {
            ans = diameter;
        }
        dfs(root.left);
        dfs(root.right);
    }

    private int depth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int lDepth = depth(root.left);
        int rDepth = depth(root.right);

        return lDepth > rDepth ? lDepth + 1 : rDepth + 1;
    }
}