543. 二叉树的直径

蔗肆稀饭呀

于 2022-03-30 13:48:52 发布

阅读量929

点赞数

分类专栏： Leetcode 文章标签：算法 c语言

本文链接：https://blog.csdn.net/weixin_48319333/article/details/123844325

版权

Leetcode 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

543. 二叉树的直径

543. 二叉树的直径 - 力扣（LeetCode） (leetcode-cn.com)

题目

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

示例 :
给定二叉树
      1
     / \
    2   3
   / \     
  4   5    
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

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

解题思路

根据王道对求树高的算法进行改进得出此解题思路

求树高递归算法：

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Ot66Q1cf-1648619227276)(C:\Users\PC\AppData\Roaming\Typora\typora-user-images\image-20220330132917140.png)]$

对以上算法进行改进,经观察发现，子树除根节点每个节点都有一个指针指向它，可将其看成路径，长度为1，到达根节点路径长度为子树高度，则有每一棵子树的直径都为左右子树的高度之和，故有：

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-s2NyZBrc-1648619286339)(C:\Users\PC\AppData\Roaming\Typora\typora-user-images\image-20220330133944041.png)]$

返回记录的最大直径即可

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int treeDepth(int*maxdiameter, struct TreeNode* root){
    if(!root)return 0;
    int left=treeDepth(maxdiameter,root->left);
    int right=treeDepth(maxdiameter,root->right);
    if(left+right>*maxdiameter)//遍历每个子树，若子树直径大于已知直径，则更新
        *maxdiameter=left+right;
    return left>right?left+1:right+1;
}
int diameterOfBinaryTree(struct TreeNode* root){
    int maxdiameter=0;
    treeDepth(&maxdiameter,root);
    return maxdiameter;

}