Cod sursa(job #2708824)

Utilizator Eva_SavaEva Maria Birsan Eva_Sava Data 19 februarie 2021 14:49:07
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.14 kb
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
vector <int> lista[100005];
int n;
int coada[100005],viz[100005];

void bfs(int nod)
{
    int sf,inc;
    for(int i=1; i<=n; i++)
        viz[i]=0;
    viz[nod]=1;
    sf=inc=1;
    coada[inc]=nod;
    while(inc<=sf)
    {
        for(int i=0; i<lista[coada[inc]].size(); i++)
        {
            if(viz[lista[coada[inc]][i]]==0)
            {
                sf++;
                coada[sf]=lista[coada[inc]][i];
                viz[lista[coada[inc]][i]]= viz[coada[inc]] + 1;
            }
        }
        inc++;
    }
}

int main()
{
    freopen("darb.in","r",stdin);
    freopen("darb.out","w",stdout);
    int i,lmax=0,f,x,y;
    scanf("%d",&n);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        lista[x].push_back(y);
        lista[y].push_back(x);
    }
    bfs(1);
    for(i=1;i<=n;i++)
        if(viz[i]>lmax)
        {
            lmax=viz[i];
            f=i;
        }
    bfs(f);
    lmax=0;
    for(i=1;i<=n;i++)
        if(viz[i]>lmax)
            lmax=viz[i];
    printf("%d",lmax);
    return 0;
}