Cod sursa(job #1190951)

Utilizator batistaUPB-Oprea-Cosmin-Dumitru batista Data 26 mai 2014 01:01:40
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.9 kb
#include <cstdio>
#include <vector>
#define N 100002
using namespace std;
vector<int>v[N];
int x, y, n, i, last, viz[N], D;
void dfs_1(int nod, int niv)
{
    viz[nod] = 1;
    for(vector<int>::iterator it=v[nod].begin(); it!=v[nod].end(); it++)
        if(!viz[*it])
            dfs_1(*it, niv+1);
    if(niv > D)
    {
        D = niv;
        last = nod;
    }
}
void dfs_2(int nod, int niv)
{
    viz[nod] = 0;
    for(vector<int>::iterator it=v[nod].begin(); it!=v[nod].end(); it++)
        if(viz[*it])
            dfs_2(*it, niv+1);
    if(niv > D)
        D = niv;
}
int main()
{
    freopen("darb.in", "r", stdin);
    freopen("darb.out", "w", stdout);
    scanf("%d",&n);
    for(i=1; i<=n-1; i++)
    {
        scanf("%d %d",&x, &y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs_1(1, 1);
    dfs_2(last, 1);
    printf("%d\n", D);
    return 0;
}