Cod sursa(job #2667744)

Utilizator george-rotariuRotariu George george-rotariu Data 3 noiembrie 2020 19:50:47
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
#include <fstream>
#include <vector>
#include <queue>
#define NMAX 100004

using namespace std;
ifstream fin ("darb.in");
ofstream fout ("darb.out");

vector <int> v[NMAX];
queue <int> q;
int dmin[NMAX];
int lgmax, n, i, x, y, nr, nod;

int main()
{
 fin>>n;
 for (i=1; i<n; i++)
     {fin>>x>>y;
      v[x].push_back(y);
      v[y].push_back(x);
     }
 q.push(1); dmin[1]=1;
 while (!q.empty())
       {
        nr=v[q.front()].size();
        for (i=0; i<nr; i++)
            if (!dmin[v[q.front()][i]])
               {
                dmin[v[q.front()][i]]=dmin[q.front()]+1;
                if (dmin[v[q.front()][i]]>lgmax)
                   {
                    lgmax=dmin[v[q.front()][i]];
                    nod=v[q.front()][i];
                   }
                q.push(v[q.front()][i]);
               }
        q.pop();
       }
 for (i=1; i<=n; i++)
     dmin[i]=0;
 dmin[nod]=1;
 q.push(nod);
 while (!q.empty())
       {
        nr=v[q.front()].size();
        for (i=0; i<nr; i++)
            if (!dmin[v[q.front()][i]])
               {
                dmin[v[q.front()][i]]=dmin[q.front()]+1;
                if (dmin[v[q.front()][i]]>lgmax) lgmax=dmin[v[q.front()][i]];
                q.push(v[q.front()][i]);
               }
        q.pop();
       }
 fout<<lgmax<<'\n';
 return 0;
}