Cod sursa(job #1380264)

Utilizator hopingsteamMatraguna Mihai-Alexandru hopingsteam Data 7 martie 2015 05:03:38
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.97 kb
#include    <iostream>
#include    <fstream>
#include    <vector>

using namespace std;

ifstream fin("darb.in");
ofstream fout("darb.out");

const int NLIM = 100005;

int N;
int maxLevel, posLevel;
vector  < int > Edge[NLIM];

bool beenThere[NLIM];

void DFS(int Node, int level)
{
    beenThere[Node] = true;
    if(level > maxLevel)
    {
        maxLevel = level;
        posLevel = Node;
    }
    for(unsigned int i = 0; i < Edge[Node].size(); i++)
    {
        int Next = Edge[Node][i];
        if(!beenThere[Next])
            DFS(Next, level + 1);
    }
}

void Reset()
{
    for(int i = 1; i <= N; i++)
        beenThere[i] = false;
}

void Read()
{
    int x, y;
    fin >> N;
    for(int i = 1; i < N; i++)
    {
        fin >> x >> y;
        Edge[x].push_back(y);
        Edge[y].push_back(x);
    }
    DFS(1,0);

    Reset();

    DFS(posLevel,0);
    fout << maxLevel + 1;
}

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