Cod sursa(job #2502316)

Utilizator NicuCNicu Capatina NicuC Data 30 noiembrie 2019 17:47:39
Problema Diametrul unui arbore Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.69 kb
#include <bits/stdc++.h>

#define NMAX 100001

using namespace std;

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

int n, a, b, maxim = -1, nmax;

set<int> m[NMAX];

set<int> temp;

void DFS(int nod, int dist, int tata) {
    if (dist > maxim) {
        maxim = dist;
        nmax = nod;
    }
    for (int i = 1; i <= n; i++)
        if (m[nod].find(i) != m[nod].end() && i != tata)
            DFS(i, dist + 1,nod);
}

int main() {
    int i;
    fin >> n;
    for (i = 1; i <= n - 1; i++) {
        fin >> a >> b;
        m[a].insert(b);
        m[b].insert(a);
    }
    DFS(1, 1,0);
    maxim = -1;
    DFS(nmax, 1,0);
    fout << maxim;
    return 0;
}