Cod sursa(job #2297269)

Utilizator cezarzbughinCezar Zbughin cezarzbughin Data 5 decembrie 2018 17:32:57
Problema Diametrul unui arbore Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.7 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("darb.in");
ofstream g("darb.out");
const int N = 100010;
int n,x,y,i,k,sol,niv[N];
vector<int> v[N];
void dfs(int);
int main()
{
    f>>n;
    for(i=1; i<n; i++)
    {
        f>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    niv[1]=1;
    dfs(1);
    for(i=1;i<=n;i++)
    {
        if(niv[i]>niv[k])
            k=i;
        niv[i]=0;
    }
    niv[k]=1;
    dfs(k);
    for(i=1;i<=n;i++)
        sol=max(sol,niv[i]);
    g<<sol;
    return 0;
}
void dfs(int nod)
{
    for(auto vec:v[nod])
        if(!niv[vec])
        {
            niv[vec]=niv[nod]+1;
            dfs(vec);
        }
}