Cod sursa(job #1205070)

Utilizator RathebaSerbanescu Andrei Victor Ratheba Data 4 iulie 2014 22:22:57
Problema Diametrul unui arbore Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.89 kb
#include <cstdio>
#include <vector>

using namespace std;
#define MAXN 100005
vector<int> lista[MAXN];
void fil(int nod);
int sum[MAXN];
int main()
{
    freopen("darb.in","r",stdin);
    freopen("darb.out","w",stdout);
    int n, x, y, i;
    scanf("%d",&n);
    for(i=1; i<n; i++)
    {
        scanf("%d%d",&x,&y);
        lista[x].push_back(y);
        lista[y].push_back(x);
    }
    fil(1);
    int max2=0, max1=0;
    for(i=0; i<lista[1].size(); i++)
    {
        if(max1 <= sum[lista[1][i]])
        {
            max2=max1;
            max1=sum[lista[1][i]];
        }
    }
    printf("%d\n",max1+max2);
    return 0;
}
void fil(int nod)
{
    int i;
    for(i=0; i<lista[nod].size(); i++)
        if(lista[nod][i] > nod)
        {
            fil(lista[nod][i]);
            sum[nod] += sum[lista[nod][i]];
        }
    sum[nod]++;
}