Cod sursa(job #1703028)

Utilizator marinaflommarina florentina marinaflom Data 16 mai 2016 00:54:51
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.15 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>

using namespace std;

vector <int> G[100010];
int viz[100010];
int N, M;
int rez;
int distantaMaxima;
int nodAflatLaDistantaMaxima;

void Read()
{
    ifstream f("darb.in");
    f >> N;
    for (int i = 1; i <= N-1; ++ i)
    {
        int x, y;
        f >> x >> y;
        G[x].push_back(y);
        G[y].push_back(x);
    }
    f.close();
}

void DFS(int node, int distanta)
{
    viz[node] = true;
    if (distanta > distantaMaxima)
    {
        distantaMaxima = distanta;
        nodAflatLaDistantaMaxima = node;
    }
    for (vector <int> :: iterator it = G[node].begin(); it != G[node].end(); ++ it)
        if (!viz[*it])
            DFS(*it, distanta + 1);
}

void Solve()
{
    distantaMaxima = -1;
    DFS(1, 0);

    for (int i = 1; i <= N; ++ i)
        viz[i] = false;
    distantaMaxima = -1;
    DFS(nodAflatLaDistantaMaxima, 0);
    rez = distantaMaxima + 1;
}

void Write()
{
    ofstream g("darb.out");
    g << rez << "\n";
    g.close();
}

int main()
{
    Read();
    Solve();
    Write();
    return 0;
}