Cod sursa(job #2134618)

Utilizator NannyiMaslinca Alecsandru Mihai Nannyi Data 18 februarie 2018 10:00:41
Problema Diametrul unui arbore Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.01 kb
#include <cstdio>
#include <vector>
#include <queue>
#include <iostream>
#define nmax 100005
using namespace std;

FILE *f=fopen("darb.in","r");
FILE *g=fopen("darb.out","w");

vector<int>Q[nmax],lastlvl;
queue<int>que;

int n,cost[nmax];
bool viz[nmax];

void read()
{
    fscanf(f,"%d",&n);
    for (int i=1; i<n; ++i)
    {
        int a,b;
        fscanf(f,"%d %d",&a,&b);
        Q[a].push_back(b);
    }
}

void dfs(int nod)
{
    viz[nod]=true;
    bool ok=0;
    for (auto w:Q[nod])
    {
        if (!viz[w])
        {
            ok=1;
            dfs(w);
            cost[nod]=max(cost[nod],cost[w]+1);
        }
    }
    if (ok==0)
        cost[nod]=1;
}

void solve()
{
    dfs(1);
    int mnd=0,mxd=0;
    for (auto w:Q[1])
        if (cost[w]>=mxd)
        {
            mnd=mxd;
            mxd=cost[w];
        }
        else if (cost[w]>mnd)
            mnd=cost[w];
    fprintf(g,"%d",mnd+mxd+1);
}

int main()
{
    read();
    solve();
    return 0;
}