Cod sursa(job #2246518)

Utilizator radugheoRadu Mihai Gheorghe radugheo Data 27 septembrie 2018 10:19:47
Problema Diametrul unui arbore Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.78 kb
#include <fstream>
#include <vector>
#define DIM 100005

using namespace std;

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

vector <int> L[DIM];
int n, i, j, p, u, nod, v[DIM], c[DIM], k;

int bfs(int start) {
    v[start] = 1;
    c[1] = start;
    int p = 1, u = 1;
    while (p <= u) {
        int nod = c[p];
        for (i=0;i<L[nod].size();i++) {
            int vecin = L[nod][i];
            if (v[vecin] == 0) {
                c[++u] = vecin;
                v[vecin] = 1;
            }
        }
        p++;
    }
    return c[u];
}


int main()
{
    fin >> n;
    for (int t=1; t<=n-1; t++){
        fin >> i >> j;
        L[i].push_back(j);
        L[j].push_back(i);
    }
    fout << bfs(1) << " " << bfs(bfs(1));
    return 0;
}