Cod sursa(job #2072817)

Utilizator GeorgeCalinPetruta George-Calin GeorgeCalin Data 22 noiembrie 2017 11:41:38
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.07 kb
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
#define nmax 100002
using namespace std;
ifstream fin("darb.in");
ofstream fout("darb.out");

vector <int> graf[nmax];
int v[nmax],x,y,n,diametru,ultimul;
bitset <nmax> viz;
queue <int> coada;

void bfs(int nod)
{
    viz[nod]=1;
    v[nod]=1;
    coada.push(nod);
    while(!coada.empty())
    {
        int nod=coada.front();
        coada.pop();
        for(int i=0;i<graf[nod].size();i++)
        {
            if(!viz[graf[nod][i]])
            {
                viz[graf[nod][i]]=1;
                v[graf[nod][i]]=v[nod]+1;
                diametru=v[graf[nod][i]];
                ultimul=graf[nod][i];
                coada.push(graf[nod][i]);
            }
        }
    }
}

int main()
{
    fin>>n;
    for(int i=1;i<=n-1;i++)
    {
        fin>>x>>y;
        graf[x].push_back(y);
        graf[y].push_back(x);
    }
    bfs(1);
    for(int i=1;i<=nmax;i++)
    {
        viz[i]=0;
        v[i]=0;
    }
    bfs(ultimul);
    fout<<diametru;
    return 0;
}