Cod sursa(job #2030071)

Utilizator patcasrarespatcas rares danut patcasrares Data 1 octombrie 2017 00:48:22
Problema Diametrul unui arbore Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.09 kb
#include<fstream>
#include<vector>
#include<iostream>
#include<queue>
#define DN 100005
using namespace std;
ifstream fin("darb.in");
ofstream fout("darb.out");
int n,x[DN],pr[DN],c[DN],ma,t,r,nod,a,b;
vector<int>v[100005];
queue<int>q;
void ve()
{
    while(!q.empty())
    {
        nod=q.front();
        q.pop();
        ma=0;
        t=0;
        for(auto i:v[nod])
            if(ma<x[i])
        {
            ma=x[i];
            t=i;
        }
        x[nod]=ma+1;
         ma=0;
        for(auto i:v[nod])
            if(ma<x[i]&&i!=t)
                ma=x[i];
        if(ma)
            r=max(r,x[nod]+ma);
        if(pr[nod])
        {
            c[pr[nod]]--;
            if(!c[pr[nod]])
                q.push(pr[nod]);
        }
    }
}
int main()
{
    fin>>n;
    for(int i=1;i<n;i++)
    {
        fin>>a>>b;
        v[a].push_back(b);
        pr[b]=a;
        c[a]++;
        //v[b].push_back(a);
    }
    for(int i=1;i<=n;i++)
        if(c[i]==0)
        {
            q.push(i);
            x[i]=1;
        }
    ve();
    fout<<r;
}