Cod sursa(job #2203102)

Utilizator YouDontNeedMyNameJurcut Paul YouDontNeedMyName Data 10 mai 2018 22:34:31
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1 kb
#include <bits/stdc++.h>
#define N_MAX 100005
using namespace std;
ifstream in("darb.in");
ofstream out("darb.out");
vector <int> lista[N_MAX];
int n,v[N_MAX];
bool ap[N_MAX];
void DFS(int k)
{
    for(int i=0; i<lista[k].size(); i++)
    {
        int vecin=lista[k][i];
        if(!v[vecin])
        {
            v[vecin]=v[k]+1;
            DFS(vecin);
        }
    }
}
void prepare()
{
    int ma=0,p=-1;
    for(int i=1; i<=n; i++)
    {
        if(v[i]>ma)
        {
            ma=v[i];
            p=i;
        }
        v[i]=0;
    }
    v[p]=1;
    DFS(p);
}
void citire()
{
    in >> n;
    for(int i=1; i<n; i++)
    {
        int x,y;
        in >> x >> y;
        lista[x].push_back(y);
        lista[y].push_back(x);
    }
}
void afis()
{
    int ma=0;
    for(int i=1; i<=n; i++)
    {
        if(v[i]>ma)
            ma=v[i];
    }
    out << ma;
}
int main()
{
    citire();
    v[1]=1;
    DFS(1);
    prepare();
    afis();
    return 0;
}