Cod sursa(job #2722636)

Utilizator AlexMariMarinescu Alexandru AlexMari Data 13 martie 2021 09:02:57
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.86 kb
#include<bits/stdc++.h>
using namespace std;
ifstream fin("darb.in");
ofstream fout("darb.out");

vector<int>g[200005];

int n,dist[200005],ans;

void dfs(int fiu,int tata)
{
    for(int i=0; i<g[fiu].size(); i++)
    {
        int nod=g[fiu][i];
        if(nod!=tata)
        {
            dist[nod]=dist[fiu]+1;
            dfs(nod,fiu);
        }
    }
}

int main()
{
    fin>>n;
    for(int i=1; i<=n-1; i++)
    {
        int x,y;
        fin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1,0);
    int maxi=0,retine=0;
    for(int i=1; i<=n; i++)
        if(dist[i]>maxi)
        {
            maxi=dist[i];
            retine=i;
        }
    for(int i=1; i<=n; i++)
        dist[i]=0;
    dfs(retine,0);
    for(int i=1; i<=n; i++)
        ans=max(ans,dist[i]);
    fout<<ans+1;
    return 0;
}