Cod sursa(job #1700040)

Utilizator zelmoatisTatuta Ionut-Catalin zelmoatis Data 9 mai 2016 10:32:54
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.83 kb
#include <iostream>
#include <vector>
#include <stack>
#include <queue>
#include <fstream>
#include <stdlib.h>
#include <time.h>
using namespace std;

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

int n,m,mx,nmx,nmx2,mid;
vector<vector<long long int> >graph;
vector<int>visited;
vector<int> parc;
stack<int>lant_max;

void bfs(int vertex)
{if(vertex<0||vertex>n-1)
    return;
queue<int> q;
unsigned int element,i;
q.push(vertex);
visited[vertex]++;
while(!q.empty())
    {element=q.front();
    for(i=0;i<graph[element].size();i++)
        if(visited[graph[element][i]]==-1)
            {q.push(graph[element][i]);
            visited[graph[element][i]]=visited[element]+1;
            parc[graph[element][i]]=element;
            }
    q.pop();
    }
}


int main()
{f>>n;
graph.resize(n);
visited.resize(n,-1);
parc.resize(n,-1);
int x,y,i;
while(f>>x>>y)
    {m++;
    x--;
    y--;
    graph[x].push_back(y);
    graph[y].push_back(x);
    }
srand(time(NULL));
x=rand()%n;
bfs(x);

mx=-1;
for(i=0;i<n;i++)
    {if(visited[i]>mx)
        {mx=visited[i];
        nmx=i;
        }
    }
/*int j;
for(i=0;i<n;i++)
    {cout<<i+1<<" ("<<visited[i]<<") "<<": ";
    for(j=0;j<graph[i].size();j++)
        {cout<<graph[i][j]+1<<" ("<<visited[graph[i][j]]<<") ";
        }
    cout<<"\n";
    }*/

for(i=0;i<n;i++)
    visited[i]=-1;
for(i=0;i<n;i++)
    parc[i]=-1;

bfs(nmx);
mx=-1;
for(i=0;i<n;i++)
    {if(visited[i]>mx)
        {mx=visited[i];
        nmx2=i;
        }
    }
g<<mx+1;
/*cout<<nmx+1<<"->"<<nmx2+1<<": "<<mx+1<<"\n";
x = nmx2;
for ( i = 0; i <= mx; i ++ )
    {cout << x+1 <<" ";
    if(i < (mx+1)/2)
        mid=x;
    x = parc[x];
    }
cout << "\n";
if( ( mx + 1 ) % 2 == 0 )
    cout << mid+1 << " " << parc[mid]+1;
else
    cout << mid+1;
*/
return 0;
}