Cod sursa(job #2048774)

Utilizator vladboss2323Ciorica Vlad vladboss2323 Data 26 octombrie 2017 16:14:09
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.3 kb
#include <iostream>
#include <vector>
#include <fstream>

using namespace std;

ifstream in("darb.in");
ofstream out("darb.out");
const int N = 100001;

vector<int> a[N];

int n,q[N],d[N],tata[N];

void citire()
{
    int x, y, m;
    in >> n  ;
    for (int i = 1; i < n; i++)
    {
        in >> x >> y;
        a[x].push_back(y);
        a[y].push_back(x);
        tata[y]=x;

    }
    in.close();
}

void bfs(int nod)
{
    int i,x,y,p=0,u=-1;
    d[nod]=0;
    q[++u]=nod;
    while(p<=u)
    {
        x=q[p++];
        for(size_t i=0; i<a[x].size(); i++)
        {
            y=a[x][i];
            if(d[y]==-1)
            {
                d[y]=1+d[x];
                q[++u]=y;
            }
        }
    }
}
int main()
{
    citire();
    int i,j,k,maxim=0;
    for(i=1; i<=n; i++)
        d[i]=-1;
    for(i=1; i<=n; i++)
        if(tata[i]==0)
            bfs(i);
    for(i=1; i<=n; i++)
        if(d[i]>maxim)
        {
            maxim=d[i];
            j=i;
        }
    for(i=1; i<=n; i++)
        d[i]=-1;
    bfs(q[n-1]);
    maxim=0;
    for(i=1; i<=n; i++)
        if(d[i]>maxim)
        {
            maxim=d[i];
            k=i;
        }
    for(i=1; i<=n; i++)
        d[i]=-1;
    bfs(j);
    out<<d[k]+1;
    return 0;
}