Cod sursa(job #3123496)

Utilizator MAlex2019Melintioi George Alexandru MAlex2019 Data 24 aprilie 2023 09:35:27
Problema Diametrul unui arbore Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.82 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

const int maxn = 1e5;
vector<int> adj[maxn + 1];
bool visited[maxn + 1];

int dfs(int node, int &last) {
    last = node;
    int ans = 0;
    for (int child: adj[node])
        if (!visited[child]) {
            visited[child] = true;
            ans = max(ans, dfs(child, last));
        }
    return ans + 1;
}

int main() {
    int n;
    fin >> n;
    for (int i = 1; i < n; i++) {
        int u, v;
        fin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    int last, aux;
    int a = dfs(1, last);
    for (int i = 1; i <= n; i++)
        visited[i] = false;
    int b = dfs(last, aux);
    fout << b << '\n';

    return 0;
}