Cod sursa(job #2226411)

Utilizator inquisitorAnders inquisitor Data 30 iulie 2018 10:42:21
Problema BFS - Parcurgere in latime Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.89 kb
#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
#define mp make_pair
#define CHECK(x) if(!(x)) return false;
typedef pair<int, int> pii;

#ifdef INFOARENA
#define ProblemName "bfs"
#endif

#define MCONCAT(A, B) A B
#ifdef ProblemName
#define InFile MCONCAT(ProblemName, ".in")
#define OuFile MCONCAT(ProblemName, ".out")
#else
#define InFile "fis.in"
#define OuFile "fis.out"
#endif

#define BUFFER_SIZE 12500000

char buffer[BUFFER_SIZE], *head = buffer, *writeHead = buffer;

bool isDigit[255];

void parseInit()
{
    buffer[fread(buffer, 1, BUFFER_SIZE, stdin)] = 0;

    for(int i = 47; i < 58; ++i) isDigit[i] = 1;
}

__attribute__((always_inline)) int get_int()
{
    int number = 0;

    for(; !isDigit[*head]; ++head);

    for(;  isDigit[*head]; ++head)

        number = number * 10 + (*head) - 48;

    return number;
}

__attribute__((always_inline)) void put_int(int x)
{
    if(x == 0)
    {
        *(writeHead++) = 48; *(writeHead++) = 32;

        return;
    }

    if(x < 0)
    {
        *(writeHead++) = 45; *(writeHead++) = 49; *(writeHead++) = 32;

        return;
    }

    char *old = writeHead;

    while(x)
    {
        *(writeHead++) = x % 10 + 48;

        x /= 10;
    }

    std::reverse(old, writeHead);

    *(writeHead++) = 32;
}

const int MAXN = 100010;
const int MAXM = 1000010;
int muchii[MAXM];
int offsetOf[MAXN];
char dst[MAXN];
int Q[MAXN], qs, qe;

void BFS(int s) {
  memset(dst, 0xFF, sizeof dst);
  qs = qe = 0;
  dst[s] = 0;
  Q[qe++] = s;
  while (qs < qe) {
    register int t = Q[qs++];
    for (register int it = offsetOf[t], tar = offsetOf[t + 1]; it != tar; ++it) {
      register int v = muchii[it];
      if (dst[v] >= 0)
        continue;
      dst[v] = dst[t] + 1;
      Q[qe++] = v;
    }
  }
}

void putNumber(int x) {
  if (x == 0) {
    *(writeHead++) = '0';
    *(writeHead++) = ' ';
    return;
  }
  if (x < 0) {
    *(writeHead++) = '-';
    *(writeHead++) = '1';
    *(writeHead++) = ' ';
    return;
  }
  char *old = writeHead;
  while (x) {
    *(writeHead++) = x % 10 + '0';
    x /= 10;
  }
  reverse(old, writeHead);
  *(writeHead++) = ' ';
}

int main() {
  freopen(InFile, "r", stdin);
  freopen(OuFile, "w", stdout);
  parseInit();
  int N = get_int(), M = get_int(), S = get_int();
  for (register int i = 0; i < M; ++i) {
    ++offsetOf[get_int()];
    get_int();
  }
  for (register int i = 1; i <= N; ++i)
    offsetOf[i] += offsetOf[i - 1];
  offsetOf[N + 1] = M;
  head = buffer;
  get_int(), get_int(), get_int();
  for (register int i = 0; i < M; ++i) {
    register int a = get_int(), b = get_int();
    muchii[--offsetOf[a]] = b;
  }
  BFS(S);
  for (register int i = 1; i <= N; ++i)
    put_int(dst[i]);
    fwrite(buffer, 1, writeHead - buffer, stdout);
  return 0;
}