Cod sursa(job #1916886)

Utilizator BlackNestaAndrei Manaila BlackNesta Data 9 martie 2017 10:36:14
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.91 kb
#include <bits/stdc++.h>
#define Nmax 100005

using namespace std;

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

int N, dp[Nmax], maxim, poz;
vector <int> L[Nmax];
bool used[Nmax];

inline void Read()
{
    int i, x, y;
    fin >> N;
    for(i = 1; i < N; i++) {
        fin >> x >> y;
        L[x].push_back(y);
        L[y].push_back(x);
    }
    fin.close();
}

inline void DFS(int nod)
{
    used[nod] = 1;
    for(auto it : L[nod]) {
        if(!used[it]) {
            dp[it] = dp[nod] +  1;
            if(dp[it] > maxim) {
                maxim = dp[it];
                poz = it;
            }
            DFS(it);
        }
    }
}

inline void Solve()
{
    int i;
    DFS(1);
    for(i = 1; i <= N; i++) {
        dp[i] = 0;
        used[i] = 0;
    }
    DFS(poz);
    fout << maxim + 1 << "\n";
}

int main()
{
    Read();
    Solve();
    return 0;
}