二叉树题目：二叉树的直径

题目

标题和出处

标题：二叉树的直径

出处：543. 二叉树的直径

难度

3 级

题目描述

要求

给定二叉树的根结点 $\text{root}$，返回其直径长度。

二叉树的直径是任意两个结点之间的最长路径长度。这条路径可能穿过也可能不穿过根结点。

两个结点之间的路径长度由它们之间边的数目表示。

示例

示例 1：

输入：$\text{root = [1,2,3,4,5]}$
输出：$\text{3}$
解释：$\text{3}$ 是路径 $\text{[4,2,1,3]}$ 或 $\text{[5,2,1,3]}$ 的长度。

示例 2：

输入：$\text{root = [1,2]}$
输出：$\text{1}$

数据范围

• 树中结点数目在范围 ${\text{[1, 10}}^{\text{4}}\text{]}$ 内
• $\text{-100}\le \text{Node.val}\le \text{100}$

解法

思路和算法

二叉树中的任意一条路径一定经过某个子树的根结点，子树可以是二叉树本身。

对于任意一个子树而言，经过该子树根结点的最长路径（以下称为「最长路径」，均指包含根结点的最长路径）一定满足以下条件：如果左子树不为空，则最长路径的左端是左子树的最深叶结点，否则最长路径的左端是根结点；如果右子树不为空，则最长路径的右端是右子树的最深叶结点，否则最长路径的右端是根结点。因此，子树的最长路径长度为该子树的左子树和右子树的深度之和，子树的深度为该子树的左子树和右子树的深度的较大值加 $1$。此处的深度定义为二叉树中结点的层数，如果二叉树为空则深度为 $0$，如果二叉树只有一个结点则深度为 $1$。

由于二叉树的最长路径长度和二叉树的深度都取决于左子树和右子树的深度，因此可以使用深度优先搜索计算二叉树的深度，计算过程中得到二叉树的直径。

计算二叉树的深度的过程是一个递归的过程，递归的终止条件是当前结点为空，此时深度为 $0$。其余情况下，首先得到当前结点的左子树和右子树的深度，然后计算以当前结点为根结点的二叉树的深度和最长路径长度，并维护二叉树的直径。遍历结束之后，即可得到二叉树的直径。

代码

class Solution {
    int diameter = 0;

    public int diameterOfBinaryTree(TreeNode root) {
        getDepth(root);
        return diameter;
    }

    public int getDepth(TreeNode node) {
        if (node == null) {
            return 0;
        }
        int leftDepth = getDepth(node.left);
        int rightDepth = getDepth(node.right);
        diameter = Math.max(diameter, leftDepth + rightDepth);
        return Math.max(leftDepth, rightDepth) + 1;
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是二叉树的结点数。每个结点都被访问一次。

• 空间复杂度：$O(n)$，其中 $n$ 是二叉树的结点数。空间复杂度主要是递归调用的栈空间，取决于二叉树的高度，最坏情况下是 $O(n)$。