Cod sursa(job #1346623)

Utilizator serban_ioan97Ciofu Serban serban_ioan97 Data 18 februarie 2015 14:25:10
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.8 kb
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>

#define nmax 100100
#define pb push_back

using namespace std;

vector<int> a[nmax];
bool used[nmax];
int i, n, m, x, y, maxd, nodem;

void dfs(int node, int lvl)
{
    vector<int>::iterator it;

    used[node]=true;

    for(it=a[node].begin(); it!=a[node].end(); it++)
    if(!used[*it]) dfs(*it, lvl+1);

    if(lvl>maxd)
    maxd=lvl, nodem=node;
}

int main()
{
    freopen("darb.in", "rt", stdin);
    freopen("darb.out", "wt", stdout);

    scanf("%d", &n);

    for(i=1; i<n; ++i)
    {
        scanf("%d%d", &x, &y);
        a[x].pb(y);
        a[y].pb(x);
    }

    dfs(1, 0);
    memset(used, false, sizeof(used));
    dfs(nodem, 0);

    printf("%d", maxd+1);

    return 0;
}