Cod sursa(job #1160274)

Utilizator ThomasFMI Suditu Thomas Thomas Data 30 martie 2014 13:42:26
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;

#define NMax 100005

ifstream f("darb.in");
ofstream g("darb.out");

int n;
int gr[NMax],gr2[NMax],viz[NMax];
vector<int> v[NMax];
queue<int> cd;

int BFS(int s)
{
    int i,mx=0,j;
    viz[s]=1;
    cd.push(s);

    while(!cd.empty())
    {
        s=cd.front(); cd.pop();
        for(i=0;i<v[s].size();i++) if(!viz[v[s][i]])
        {
            viz[v[s][i]]=1;
            gr[v[s][i]]=gr[s]+1;
            cd.push(v[s][i]);
            if(mx<gr[v[s][i]]) mx=gr[v[s][i]], j=v[s][i];
        }
    }

    return j;
}

int BFS2(int s)
{
    int i,mx=0;
    viz[s]=0;
    gr2[s]=1;
    cd.push(s);

    while(!cd.empty())
    {
        s=cd.front(); cd.pop();
        for(i=0;i<v[s].size();i++) if(viz[v[s][i]])
        {
            viz[v[s][i]]=0;
            gr2[v[s][i]]=gr2[s]+1;
            cd.push(v[s][i]);
            if(mx<gr2[v[s][i]]) mx=gr2[v[s][i]];
        }
    }

    return mx;
}

int main()
{
    int i,a,b;

    f>>n;

    for(i=1;i<=n;i++)
    {
        f>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }

    g<<BFS2(BFS(1))<<"\n";

    f.close();
    g.close();
    return 0;
}