Cod sursa(job #2868168)

Utilizator Florian11232Florian Susai Florian11232 Data 10 martie 2022 19:30:05
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.42 kb
#include <iostream>
#include <queue>
#include <vector>
#include <fstream>
#define INF 1000005
#define dim 100001
using namespace std;

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

vector <int> graf[dim];
queue <int> q;
int d[dim];

int n;

void bfs(int s) {
    int x, y;
    for(int i=1; i<=n;i++) {
        d[i]=INF;
    }
    d[s]=0;
    q.push(s); // inseram in coada nodul de start
    
    while(!q.empty()) {
        // extrag nodul de la inceputul cozii
        x = q.front();
        q.pop(); // si elimin nodul

        // ma uit la vecinii lui x
        for (int i = 0; i < graf[x].size(); i++) {
            y = graf[x][i];

            if (d[y] > d[x] + 1) {
                d[y] = d[x] + 1; // fac update in vectorul d
                q.push(y);
            }
        }
    }
}


int main()
{
    int s, x, y, dmax , nmax ;
    in >> n;
    for(int i=1; i <= n-1; i++){
        in >> x >> y;
        //adaug in lista nodlui x pe y nu invesr pt ca e orientat
        graf[x].push_back(y);
        graf[y].push_back(x);
    }
    //out << '\n';
    bfs(1);
    dmax = d[1];
    nmax = 1;
    for(int i=2; i <= n;i++){
        if(d[i] > dmax){
            dmax = d[i];
            nmax = i;
        }
    }
    //out<<nmax<<endl;
    bfs(nmax);
    int maxi=d[1];
    for(int i=2; i <= n;i++){
        maxi = max(maxi,d[i]);
    }
    out<<maxi+1;
    return 0;
}