Cod sursa(job #2425260)

Utilizator stan_flaviusStan Flavius Stefan stan_flavius Data 24 mai 2019 17:38:52
Problema Diametrul unui arbore Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 kb
#include <fstream>
#define nmax 100001
#include <vector>

using namespace std;
ifstream fin("p.in");
ofstream fout("p.out");

vector<int> v[nmax];
int n;

void read()
{
    int i;
    fin>>n;
    for(i=1; i<n; i++)
        {
            int x,y;
            fin>>x>>y;
            v[x].push_back(y);
            v[y].push_back(x);
        }
}

int viz1[nmax];
void dfs(int x,int y)
{
    viz1[x]=viz1[y]+1;
    for(int i=0; i<v[x].size(); i++)
        if(!viz1[v[x][i]]) dfs(v[x][i],x);
}

int main()
{
    read();
    dfs(1,0);
    int mx=0,val;
    for(int i=1; i<=n; i++)
        {
            if(viz1[i]>mx) {mx=viz1[i]; val=i;}
            viz1[i]=0;
        }
    dfs(val,0);
    for(int i=1; i<=n; i++)
        {
            if(viz1[i]>mx) {mx=viz1[i]; val=i; viz1[i]=0;}
        }
    fout<<mx;
    return 0;
}