Cod sursa(job #2968809)

Utilizator Samoila_AlexandruSamoilaAlexandru Samoila_Alexandru Data 22 ianuarie 2023 00:03:30
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.94 kb
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

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

const int nMax=1e5+5;

int n, d[nMax], nod, diametru;
vector<int> g[nMax];

void bfs(int k)
{
    queue<int> q;
    q.push(k);
    d[k]=1;
    diametru=d[k];
    while(!q.empty())
    {
        int k=q.front();
        q.pop();

        for(auto i: g[k])
            if(!d[i])
        {
            d[i]=d[k]+1;
            q.push(i);
            nod=i;
            diametru=max(diametru, d[i]);
        }
    }
}

int main()
{
    ios_base::sync_with_stdio(false);
    fin.tie(0);

    fin>>n;
    int x, y;
    for(int i=1; i<n; i++)
    {
        fin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    fin.close();

    bfs(1);

    for(int i=1; i<=n; i++)
        d[i]=0;

    bfs(nod);

    fout<<diametru;

    fout.close();

    return 0;
}