Cod sursa(job #2595479)

Utilizator etienAndrone Stefan etien Data 7 aprilie 2020 19:55:34
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.78 kb
#include<fstream>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
ifstream fin("darb.in");
ofstream fout("darb.out");
vector<int>v[100001];
int n,i,a,b,d[100001],vfmax,vecin,k;
bool viz[100001];
void dfs(int vf)
{
    for(int i=0; i<v[vf].size(); i++)
    {
        vecin=v[vf][i];
        if(!viz[vecin])
        {
            viz[vecin]=true;
            d[vecin]=d[vf]+1;
            dfs(vecin);
        }
    }
    if(d[vfmax]<d[vf])
    {
        vfmax=vf;
    }

}
int main()
{
    fin>>n;
    viz[1]=true;
    for(i=1; i<n; i++)
    {
        fin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    dfs(1);
    d[vfmax]=0;
    k=vfmax;
    fill(viz,viz+100001,0);
    dfs(k);
    fout<<d[vfmax]+1;
}