Cod sursa(job #2615021)

Utilizator paul3ioanCirstean Paul Ioan paul3ioan Data 13 mai 2020 14:58:29
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.67 kb
#include <fstream>
#include <vector>
#include <cstring>

using namespace  std;
const int N = 100005;
vector <int > L[N];
int ans, x , y , n;
int node, viz[N], maxim,dist;
ifstream cin("darb.in");
ofstream cout("darb.out");
void dfs(int x)
{
    viz[x] = 1;
    for(int i : L[x])
        if(!viz[i])
            dist++, dfs(i), dist--;
    if(L[x].size()== 1 and maxim < dist)
    {
        node = x;
        maxim = dist;
    }
}
int main() {
    cin >> n;
    for(int i =1 ;i <= n - 1 ;i ++)
    {
        cin >> x >> y;
        L[x].push_back(y);
        L[y].push_back(x);
    }
    dfs(1);
    memset(viz, 0 , sizeof(viz));
    dfs(node);
    cout << maxim + 1;
    return 0;
}