Cod sursa(job #1922228)

Utilizator nicu_serteSerte Nicu nicu_serte Data 10 martie 2017 16:32:58
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.22 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("darb.in");
ofstream fout("darb.out");
#define nmax 100001
vector <int> g[nmax];
int n, d[nmax];
void citire()
{
    int i, j, m;
    fin>>n;
    for(m=1; m<=n; m++)
    {
        fin>>i>>j;
        g[i].push_back(j);
        g[j].push_back(i);
    }
    fin.close();
}
void bfs(int nod)
{
    queue <int> q;
    int k;
    bool viz[nmax]={0};
    vector <int>::iterator it;
    q.push(nod);
    viz[nod]=1;
    d[nod]=1;
    while(!q.empty())
    {
        k=q.front();
        q.pop();
        for(it=g[k].begin(); it!=g[k].end(); it++)
            if(!viz[*it])
            {
                d[*it]=d[k]+1;
                viz[*it]=1;
                q.push(*it);
            }
    }
}
int main()
{
    int i, dmax=0, x=0;
    citire();
    bfs(1);
    for(i=1; i<=n; i++)
        if(dmax<d[i])
        {
            dmax=d[i];
            x=i;
        }
    for(i=1; i<=n; i++)
        d[i]=0;
    bfs(x);
    dmax=0; x=0;
    for(i=1; i<=n; i++)
        if(dmax<d[i])
        {
            dmax=d[i];
            x=i;
        }
    fout<<d[x]<<'\n';
    fout.close();
    return 0;
}