Cod sursa(job #3326181)

Utilizator itachi_uchihaDarius itachi_uchiha Data 27 noiembrie 2025 17:15:58
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.88 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>

using namespace std;

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

const int N = 100001;

vector<int> g[N];
int vizitat[N];
int sol;

int dp[N];

void dfs(int nod){
    vizitat[nod] = 1;
    int fiu1 = 0;
    int fiu2 = 0;
    dp[nod] = 1;
    for (int x : g[nod])
    {
        if(!vizitat[x]){
            dfs(x);
            dp[nod] = max(dp[nod], dp[x] + 1);
            if(dp[x] > fiu1){
                fiu2 = fiu1;
                fiu1 = dp[x];
            }
            else if(dp[x] > fiu2){
                fiu2 = dp[x];
            }
        }
    }

    sol = max(sol, fiu1+fiu2 + 1);
}

int main (){
    int n;
    fin >> n;

    for (int i = 1; i <= n-1; i++){
        int x, y;
        fin >> x >> y;
        g[x].push_back(y);
        g[y].push_back(x);
    }

    dfs(1);

    fout << sol;
}