Cod sursa(job #1290074)

Utilizator gapdanPopescu George gapdan Data 10 decembrie 2014 19:50:48
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.4 kb
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#define NMAX 100005
using namespace std;
vector<int>v[NMAX];
queue<int>q;
int tata[NMAX],d[NMAX],viz[NMAX];
int n,m,i,poz,last,x,y;
void BFS(int nod)
{
    q.push(nod);
    viz[nod]=1;
    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        last=x;
        vector<int>::iterator it;
        for (it=v[x].begin();it!=v[x].end();++it)
        {
            if (viz[*it]==0)
            {
                q.push(*it);
                viz[*it]=1;
            }
        }
    }
}
void DFS(int nod)
{
    vector<int>::iterator it;
    viz[nod]=1;
    for (it=v[nod].begin();it!=v[nod].end();++it)
    {
        if (viz[*it]==0)
        {
            d[*it]=d[nod]+1;
            tata[*it]=nod;
            DFS(*it);
        }
    }
}
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);
        }
    BFS(1);
    memset(viz,0,sizeof(viz));
    d[last]=1;
    DFS(last);
    int Max=0;
    for (i=1;i<=n;++i)
        if (d[i]>Max) {Max=d[i];poz=i;}
    printf("%d\n",Max);
   /* while(tata[poz]!=0)
    {
        printf("%d ",poz);
        poz=tata[poz];
    }
    printf("%d\n",poz);*/
    return 0;
}