Cod sursa(job #2846272)

Utilizator cdenisCovei Denis cdenis Data 9 februarie 2022 02:13:48
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 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;
vector < int > v[MAX];

void DFS(int nod, int tata)
{
    int m1=0,m2=0;
    for(auto vec : v[nod])
        if(vec!=tata)
        {
            DFS(vec,nod);
            if(viz[vec]>m1)
            {
                m2=m1;
                m1=viz[vec];
            }
            else if(viz[vec]>m2)
                m2=viz[vec];
            viz[nod]=max(viz[nod],viz[vec]+1);
        }
    lant=max(lant,m1+m2);
    if(!viz[nod])
        viz[nod]=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);
    fout << lant+1;
    return 0;
}