Cod sursa(job #2846273)

Utilizator cdenisCovei Denis cdenis Data 9 februarie 2022 02:18:14
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.67 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

const int MAX=1e5+5;
int n,a,b,viz[MAX],lant,fr;
vector < int > v[MAX];

void DFS(int nod, int niv)
{
    viz[nod]=1;
    if(niv>lant)
    {
        lant=niv;
        fr=nod;
    }
    for(auto vec : v[nod])
        if(!viz[vec])
            DFS(vec,niv+1);
}

int main()
{
    fin >> n;
    for(int i=1;i<n;i++)
    {
        fin >> a >> b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    DFS(1,0);
    for(int i=1;i<=n;i++)
        viz[i]=0;
    DFS(fr,0);
    fout << lant+1;
    return 0;
}