Cod sursa(job #1373446)

Utilizator RathebaSerbanescu Andrei Victor Ratheba Data 4 martie 2015 18:47:52
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.96 kb
#include <cstdio>
#include <vector>

using namespace std;
#define MAX 100005

void dfs(int nod);

vector<int> v[MAX];
int n, viz[MAX];

int main()
{
    freopen("darb.in", "r", stdin);
    freopen("darb.out", "w", stdout);
    int i, a, b, ultnod, sol;
    scanf("%d", &n);
    for(i=1; i<n; i++)
    {
        scanf("%d%d", &a, &b);
        v[a].push_back(b);
        v[b].push_back(a);
    }
    dfs(1);
    ultnod = 1;
    for(i=2; i<=n; i++)
        if(viz[ultnod] < viz[i])
            ultnod = i;
    for(i=1; i<=n; i++) viz[i] = 0;
    dfs(ultnod);
    sol = 0;
    for(i=1; i<=n; i++)
        if(sol < viz[i])
            sol = viz[i];
    printf("%d\n", sol+1);
    return 0;
}
void dfs(int nod)
{
    int fiu;
    vector<int>::iterator it;
    for(it=v[nod].begin(); it!=v[nod].end(); ++it)
    {
        fiu = *it;
        if(!viz[fiu])
        {
            viz[fiu] = viz[nod] + 1;
            dfs(fiu);
        }
    }
}