Cod sursa(job #1650510)

Utilizator george_stelianChichirim George george_stelian Data 11 martie 2016 18:52:23
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.93 kb
#include <cstdio>
#include <vector>
#include <queue>

using namespace std;

vector<int> v[100010];
queue<int> q;
int d[100010],n;

void bfs(int nod)
{
    for(int i=1;i<=n;i++) d[i]=0;
    d[nod]=1;
    q.push(nod);
    while(!q.empty())
    {
        int nod=q.front();q.pop();
        for(vector<int>::iterator it=v[nod].begin();it!=v[nod].end();it++)
            if(!d[*it])
            {
                d[*it]=d[nod]+1;
                q.push(*it);
            }
    }
}

int main()
{
    freopen("darb.in", "r", stdin);
    freopen("darb.out", "w", stdout);
    int x,y,sol=0,nod=0;
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    bfs(1);
    for(int i=1;i<=n;i++)
        if(d[i]>d[nod]) nod=i;
    bfs(nod);
    for(int i=1;i<=n;i++)
        sol=max(sol,d[i]);
    printf("%d",sol);
    return 0;
}