【LeetCode每日一题】——543.二叉树的直径

一【题目类别】

• 树

二【题目难度】

• 简单

三【题目编号】

• 543.二叉树的直径

四【题目描述】

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

五【题目示例】

• 示例 :

  • 给定二叉树

  • 

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

六【解题思路】

• 要注意理解题目中的“直径”的概念，题目的意思是任意两个点之间的最长路径就是“直径”
• 那么我们只需要对本棵树进行深度优先搜索（dfs），计算每个结点的左右子树的长度，如果左右子树的长度之和大于之前保存的“直径”，说明需要更新“直径”
• 说白了这道题就是在求树的深度的基础上，加了一个求任意两节点之间最长路径的问题
• 从最小的节点往整棵树遍历
• 最后返回结果即可

七【题目注意】

• 两结点之间的路径长度是以它们之间边的数目表示。

八【时间频度】

• 时间复杂度：$O(n)$，其中$n$为树的节点个数
• 空间复杂度：$O(h)$，其中h为树的高度，也就是递归栈的深度

九【代码实现】

1. Java语言版

class Solution {

    int res = 0;

    public int diameterOfBinaryTree(TreeNode root) {
        dfsDiameter(root);
        return res;
    }

    public int dfsDiameter(TreeNode root){
        if(root == null){
            return 0;
        }
        int left = dfsDiameter(root.left);
        int right = dfsDiameter(root.right);
        res = Math.max(res,left + right);
        return Math.max(left,right) + 1;
    }

}

2. C语言版

int dfsDiameter(struct TreeNode* root,int* res)
{
    if(root == NULL)
    {
        return 0;
    }
    int left = dfsDiameter(root->left,res);
    int right = dfsDiameter(root->right,res);
    *res = fmax(*res,left + right);
    return fmax(left,right) + 1;
}

int diameterOfBinaryTree(struct TreeNode* root)
{
    int res = 0;
    dfsDiameter(root,&res);
    return res;
}

十【提交结果】

1. Java语言版
   

2. C语言版