Cod sursa(job #2226497)

Utilizator inquisitorAnders inquisitor Data 30 iulie 2018 11:47:01
Problema BFS - Parcurgere in latime Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 2.87 kb
#include <bits/stdc++.h>

/*
#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;
}
*/

const int MAXBUF = 20000000;
char buffer[MAXBUF];
char *head;
int isDigit[255];
char *writeHead;

void parseInit()
{
    int a = fread(buffer, 1, MAXBUF, stdin);

    buffer[a] = 0;

    head = buffer;

    memset(isDigit, 0, sizeof isDigit);

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

    writeHead = head;
}

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

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

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

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

    return number;
}

char outBuffer[20000000]; unsigned int p = ~0x0;

__attribute__((always_inline)) void put_int(unsigned int x)
{
    if(x < 0)
    {
        outBuffer[++p] = 45; outBuffer[++p] = 49; outBuffer[++p] = 32;

        return;
    }

    unsigned int digits = x > 0x3B9AC9FF ? 0xB :
                          x > 0x5F5E0FF  ? 0xA :
                          x > 0x98967F   ? 0x9 :
                          x > 0xF423F    ? 0x8 :
                          x > 0x1869F    ? 0x7 :
                          x > 0x270F     ? 0x6 :
                          x > 0x3E7      ? 0x5 :
                          x > 0x63       ? 0x4 :
                          x > 0x9        ? 0x3 : 0x2;

    for(unsigned int i = digits; --i; x /= 0xA)
    {
        outBuffer[p + i] = x % 0xA + 0x30;
    }

    outBuffer[p = p + digits] = 0x20;
}


int muchii[1000010];
int degree[100010];
char level[100010];

void BFS(int s)
{
    int Q[100010], qs = 0, qe = 0;

    memset(level, 0xFF, sizeof level);

    level[s] = 0;

    Q[qe++] = s;

    while(qs < qe)
    {
        int t = Q[qs++];

        for(int it = degree[t], tar = degree[t + 1]; it != tar; ++it)
        {
            int v = muchii[it];

            if(level[v] >= 0) continue;

            level[v] = level[t] + 1;

            Q[qe++] = v;
        }
    }
}

int main()
{
    freopen("bfs.in", "r", stdin);
    freopen("bfs.out", "w", stdout);

    parseInit();

    int N = get_int(), M = get_int(), start = get_int();

    for(int i = 0; i < M; ++i)
    {
        ++degree[get_int()];

        get_int();
    }

    for(int i = 1; i <= N; ++i)
    {
        degree[i] += degree[i - 1];
    }

    degree[N + 1] = M;

    head = buffer;

    get_int(), get_int(), get_int();

    for(int i = 0; i < M; ++i)
    {
        int u = get_int(), v = get_int();

        muchii[--degree[u]] = v;
    }

    BFS(start);

    for(int i = 1; i <= N; ++i)
    {
        put_int(level[i]);
    }

    fwrite(outBuffer, 1, p, stdout);

    return 0;
}