Cod sursa(job #2325271)

Utilizator aturcsaTurcsa Alexandru aturcsa Data 22 ianuarie 2019 13:42:35
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.69 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("darb.in");
ofstream fout("darb.out");
vector <int> A[100001];
bool viz[100001];
int n,maxi,locmax;
void DFS(int nod,int d)
{
    viz[nod]=1;
    if(d>maxi)
    {
        maxi=d;
        locmax=nod;
    }
    for(auto it:A[nod])
    {
        if(viz[it]==0)
        {
            DFS(it,d+1);
        }
    }
}
int main()
{
    fin>>n;
    for(int i=1;i<=n-1;i++)
    {
        int a,b;
        fin>>a>>b;
        A[a].push_back(b);
        A[b].push_back(a);
    }
    DFS(1,0);
    for(int i=1;i<=n;i++)
        viz[i]=0;
    DFS(locmax,1);
    fout<<maxi;
    return 0;
}