Cod sursa(job #2204996)

Utilizator andreistanStan Andrei andreistan Data 17 mai 2018 16:19:49
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.88 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream f("darb.in");
ofstream g("darb.out");

vector<int> v[100001];
int N, nodmax;
bool viz[100001];
int lg[100001], Coada[100001];

void BFS(int x)
{
    int p = 1, u = 1;
    Coada[1] = x;
    viz[x] = 1;
    while(p <= u)
    {
        int crt = Coada[p++];
        for(auto &it : v[crt])
            if(viz[it] == 0)
            {
                Coada[++u] = it;
                viz[it] = 1;
                lg[it] = 1 + lg[crt];
                nodmax = it;
            }
    }
}

int main()
{
    int x, y;
    f >> N;
    for(int i = 1; i < N; i++)
    {
        f >> x >> y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    BFS(1);
    for(int i = 1; i <= N; i++)
        viz[i] = 0,lg[i]=1;
    BFS(nodmax);
    g << lg[nodmax];
    return 0;
}