Cod sursa(job #2204284)

Utilizator nurof3nCioc Alex Andrei nurof3n Data 15 mai 2018 12:04:37
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

const int NMAX = 100001;
int N, ult, dist[NMAX], coada[NMAX];
bool viz[NMAX];
vector<int> lista[NMAX];

int BFS(int x)
{
    int st = 0, dr = 1;
    viz[x] = true;
    coada[++st] = x;
    while(st <= dr)
    {
        int crt = coada[st++];
        for(auto &it : lista[crt])
            if(viz[it] == false)
            {
                coada[++dr] = it;
                dist[it] = dist[crt] + 1;
                viz[it] = true;
            }
    }
    return coada[dr];
}
int main()
{
    int x, y;
    f >> N;
    for(int i = 1; i < N; i++)
    {
        f >> x >> y;
        lista[x].push_back(y);
        lista[y].push_back(x);
    }
    ult = BFS(1);
    for(int i = 1; i <= N; i++)
        viz[i] = false, dist[i] = 1;
    g << dist[BFS(ult)];
    return 0;
}