Cod sursa(job #3137778)

Utilizator SergetecLefter Sergiu Sergetec Data 14 iunie 2023 21:34:24
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.87 kb
/*
 * Lefter Sergiu - 14/06/2023
*/
#include <fstream>
#include <vector>

using namespace std;

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

const int N = 1e5;

int n, d1[N + 1], dp[N + 1];
vector <int> a[N + 1];

void dfs(int x, int t, int d[]) {
    d[x] = 1 + d[t];
    for (auto y: a[x]) {
        if (d[y] == 0) {
            dfs(y, x, d);
        }
    }
}

int maxim(int v[], int n) {
    int pmax = 1;
    for (int i = 2; i <= n; i++) {
        if (v[i] > v[pmax]) {
            pmax = i;
        }
    }
    return pmax;
}

int main() {
    in >> n;
    for (int i = 1; i < n; i++) {
        int x, y;
        in >> x >> y;
        a[x].push_back(y);
        a[y].push_back(x);
    }
    dfs(1, 0, d1);
    int p = maxim(d1, n);
    dfs(p, 0, dp);
    out << dp[maxim(dp, n)];
    in.close();
    out.close();
    return 0;
}