scau18923

ko_he

于 2024-06-10 15:45:07 发布

阅读量233

点赞数 4

文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/ko_he/article/details/139577548

版权

18923 二叉树的直径

时间限制:1000MS 代码长度限制:10KB
提交次数:0 通过次数:0

题型: 编程题语言: G++;GCC

Description

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
          1
         / \
        2   3
       / \     
      4   5   
答案为3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

输入格式

共n行。
第一行一个整数n，表示有n个结点，编号为1至n。
第二行至第n行，每行有两个整数x和y，表示在二叉树中x为y的父节点。x第一次出现时y为左孩子

输出格式

输出二叉树的直径。

输入样例

想法：求出左子树深度，再求出右子树深度，相加最大的和为二叉树直径；

每个节点遍历一遍，时间复杂度大约为O(4n)

#include <iostream>
#include <cstdlib>

using namespace std;

int max_deepth;//用于保存左/右子树的深度
int max_val;
int n;

struct LNode{
    int l_child;
    int r_child;
};

void check(int now,int deepth,LNode* &head)
{
    if(now==0)
    {
        max_deepth=max(max_deepth,deepth);
        return;
    }
    else
    {
        check(head[now].l_child,deepth+1,head);
        check(head[now].r_child,deepth+1,head);
    }
}

void create(int x,int y,LNode* &head)
{
    if(head[x].l_child==0)
    {
        head[x].l_child=y;
    }
    else
    {
        head[x].r_child=y;
    }
}


int main()
{
    LNode* head;
    scanf("%d",&n);
    head=(LNode*)malloc(sizeof(LNode)*(n+1));//节点0到n,其中节点0用于返回

    int i;
    int x,y,r,l;

    for(i=0;i<=n;i++)//全部初始化
    {
        head[i].r_child=0;
        head[i].l_child=0;
        head[i].tag=0;
    }




    for(i=1;i<=n-1;i++)
    {
        scanf("%d %d",&x,&y);
        create(x,y,head);
    }

    for(i=1;i<=n;i++)
    {
        check(head[i].l_child,0,head);
        l=max_deepth;//记录左子树深度
        max_deepth=0;//一定要复位
        check(head[i].r_child,0,head);
        r=max_deepth;//记录右子树深度
        max_deepth=0;//复位
        max_val=max(max_val,r+l);

    }
    printf("%d",max_val);

}