Cod sursa(job #1435445)

Utilizator zpaePopescu Andreea zpae Data 13 mai 2015 10:41:54
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.28 kb
#include <cstdio>
#include <vector>
#include <deque>
using namespace std;
int n,u[100007],k,kp;

int main()
{
    freopen("darb.in","r",stdin);
    freopen("darb.out","w",stdout);
    int i,x,y,f;
    vector <int> v[100007];
    deque <int> q;
    scanf("%d",&n);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    q.push_back(1);
    u[1]=-1;
    while(!q.empty())
    {
        f=q.front();
        for(i=0;i<v[f].size();i++)
            if(!u[v[f][i]])
            {
                u[v[f][i]]=u[f]-1;
                if(k>u[v[f][i]])
                {
                    k=u[v[f][i]];
                    kp=v[f][i];
                }
                q.push_back(v[f][i]);
            }
        q.pop_front();
    }
    q.push_back(kp);
    u[kp]=1;
    k=0;
    while(!q.empty())
    {
        f=q.front();
        for(i=0;i<v[f].size();i++)
            if(u[v[f][i]]<0)
            {
                u[v[f][i]]=u[f]+1;
                if(k<u[v[f][i]])
                {
                    k=u[v[f][i]];
                    kp=v[f][i];
                }
                q.push_back(v[f][i]);
            }
        q.pop_front();
    }
    printf("%d\n",k);
    return 0;
}