Cod sursa(job #2862711)

Utilizator alexmorosanuMorosanu Alexandru alexmorosanu Data 5 martie 2022 19:11:09
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.19 kb
#include <fstream>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
ifstream f("darb.in");
ofstream g("darb.out");
vector <int> v[100011];
queue <int> q;
int n,i,x,y,fost[100011],nr,k,K,M;
int main()
{
    f>>n;
    for(i=1;i<n;i++)
    {
        f>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    q.push(1);
    fost[1]=nr=1;
    while(q.empty()==false)
    {
        k=q.front();
        nr++;
        q.pop();
        if(fost[k]>M)
        {
            M=fost[k];
            K=k;
        }
        for(i=0;i<v[k].size();i++)
            if(fost[v[k][i]]==0)
            {
                fost[v[k][i]]=fost[k]+1;
                q.push(v[k][i]);
            }
    }
    memset(fost,0,sizeof(fost));
    fost[K]=nr=1;
    q.push(K);
    while(q.empty()==false)
    {
        k=q.front();
        nr++;
        q.pop();
        if(fost[k]>M)
        {
            M=fost[k];
            K=k;
        }
        for(i=0;i<v[k].size();i++)
            if(fost[v[k][i]]==0)
            {
                fost[v[k][i]]=fost[k]+1;
                q.push(v[k][i]);
            }
    }
    g<<M;
    return 0;
}