Cod sursa(job #1041368)

Utilizator bghimisFMI Ghimis Bogdan bghimis Data 25 noiembrie 2013 19:31:22
Problema BFS - Parcurgere in latime Scor 80
Compilator cpp Status done
Runda Arhiva educationala Marime 1.39 kb
#include <fstream>
#include <queue>
#include <vector>
#include <bitset>
#include <utility>
#include <algorithm>
using namespace std;

ifstream in ("bfs.in");
ofstream out("bfs.out");

bitset<100005> vizitat;

bool sortPair (pair<int, int> first,
	       pair<int, int> second)
{
  if (first.first < second.first)
    return true;
  return false;
}

void bfs (int nod, vector<int> *relatii)
{
  vector<pair<int, int> > myHash;
  queue<pair<int, int> > myQueue;
  myQueue.push(make_pair(nod, 0));
  vizitat[nod] = true;

  while (!myQueue.empty())
    {
      pair<int, int> current = myQueue.front(); 
      int nod_curent = current.first;
      int valoare_curenta = current.second;

      myHash.push_back(current);
      
      for (auto &it : relatii[nod_curent])
	if (vizitat[it] == false)
	  myQueue.push (make_pair(it, valoare_curenta + 1)), vizitat[it] = true;

      myQueue.pop();
    }

  sort (myHash.begin(), myHash.end(), sortPair);
  int i = 1;
  for (vector<pair<int, int> >::iterator it = myHash.begin(); it != myHash.end(); ++it, ++i)
    {
      while (i++ != (*it).first)
	out << -1 << " ";
      --i;

      out << (*it).second << " ";
    }
}

int main()
{
  int n; in >> n;
  int m; in >> m;
  int nod_plecare; in >> nod_plecare;

  vector<int> relatii[100005];
  for (int i = 0; i < m; ++i)
    {
      int x, y; in >> x >> y;
      relatii[x].push_back(y);
    }

  bfs (nod_plecare, relatii);
}