Cod sursa(job #2259203)

Utilizator TavinciStefanescu Octavian Tavinci Data 13 octombrie 2018 10:23:20
Problema Diametrul unui arbore Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.94 kb
#include <bits/stdc++.h>
#define NMAX 100001

using namespace std;

vector<int> G[NMAX];

int n, m, viz[NMAX], comp, cost[NMAX], maxim=0, nodPornire;
ifstream fin("darb.in");
ofstream fout("darb.out");

void DFS(int nod)
{
    viz[nod]=1;
    int nrNod=G[nod].size();
    int actualNod;
    for(int i=0; i<nrNod; i++)
    {
        actualNod=G[nod][i];
        if(viz[actualNod]==0)
        {
            cost[actualNod]=cost[nod]+1;
            if(cost[actualNod]>maxim)
            {
                nodPornire=actualNod;
                maxim=cost[actualNod];
            }
            DFS(actualNod);
        }
    }
}

int main()
{
    fin>>n;
    int p1, p2;
    for(int i=1; i<=n-1; i++)
    {
        fin>>p1>>p2;
        G[p1].push_back(p2);
        G[p2].push_back(p1);
    }

    DFS(1);
    for(int i=1;i<=n;i++)
        viz[i]=0;
    DFS(nodPornire);
    fout<<cost[nodPornire];

    return 0;
}