Cod sursa(job #1131148)

Utilizator MarghescuGabriel Marghescu Marghescu Data 28 februarie 2014 18:07:34
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.2 kb
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#define nmax 100002
using namespace std;

queue <int> q;
int viz[nmax];
struct nod{
    int v;
    nod *urm;
};
nod *LA[nmax],*p;

int bfs(int x)
{
    memset(viz,0,sizeof(viz));
    q.push(x);
    viz[x]=1;
    int maxim=x;
    while(!q.empty())
    {
        int y=q.front();

        //for(int i=0;i<v[y].size();i++)
        for(nod* p=LA[y];p;p=p->urm)
        {
            int z=p->v;

            if(viz[z]==0)
            {
                q.push(z);
                viz[z]=1+viz[y];
                if(viz[z]>viz[maxim])
                    maxim=z;
            }
        }
        q.pop();
    }
    return maxim;

}
int main()
{
    freopen("darb.in","r",stdin);
    freopen("darb.out","w",stdout);
    int x,y,i,n;
    scanf("%d",&n);
    for(i=1;i<n;i++)
    {
        scanf("%d %d",&x,&y);
        //v[x].push_back(y);
        //v[y].push_back(x);
        p=new nod;
        p->v=y;
        p->urm=LA[x];
        LA[x]=p;
        p=new nod;
        p->v=x;
        p->urm=LA[y];
        LA[y]=p;

    }

    int drum=bfs(bfs(1));
    printf("%d",viz[drum]);
    return 0;
}