Cod sursa(job #1379126)

Utilizator roxana_97Soare Roxana Florentina roxana_97 Data 6 martie 2015 16:27:26
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include<fstream>
#include<vector>
#include<bitset>
#include<queue>
#define Nmax 100009
using namespace std;

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

int x,y,n,m,last,D,d[Nmax];
vector<int> G[Nmax];
bitset<Nmax> viz;
queue<int> Q;

inline void BFS(int node)
{
    for(int i=1;i<=n;i++) viz[i]=d[i]=0;
    viz[node]=d[node]=1;
    Q.push(node);
    while(!Q.empty())
    {
        node=Q.front();
        Q.pop();
        for(vector<int> :: iterator it=G[node].begin();it!=G[node].end();it++)
            if(!viz[*it])
            {
                viz[*it]=1;
                last=*it;
                D=d[*it]=d[node]+1;
                Q.push(*it);
            }
    }
}

int main()
{
    f>>n;
    for(int i=1;i<n;i++)
    {
        f>>x>>y;
        G[x].push_back(y);
        G[y].push_back(x);
    }
    BFS(1);
    BFS(last);
    g<<D<<'\n';
    f.close();g.close();
    return 0;
}