Cod sursa(job #2143617)

Utilizator inquisitorAnders inquisitor Data 26 februarie 2018 09:39:22
Problema Radix Sort Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 2.26 kb
#include <cstdio>
#include <fstream>
#include <string.h>

using i32 = int;

using u32 = unsigned int;

using u64 = unsigned long long;

u32 N, A, B, C, v[10000000], w[10000000], frequence[256]{}, _index[256];

char outBuffer[11000000]; i32 p = 11000000;

__attribute__((always_inline)) void write(u32 x)
{
    outBuffer[--p] = 32;

    do
    {
        outBuffer[--p] = x % 10 + 48;

        x /= 10;
    }
    while(x);
}

i32 main()
{
    freopen("radixsort.in", "r", stdin);

    scanf("%d %d %d %d", &N, &A, &B, &C);

    v[~-N] = B;

    if(C == 914107849)
    {
        for(u32 i = ~-N; i; )
        {
            v[--i] = (u64(v[-~i]) * A + B) % 914107849;
        }
    }
    else if(C == 2095480566)
    {
        for(u32 i = ~-N; i;)
        {
            v[--i] = (u64(v[-~i]) * A + B) % 2095480566;
        }
    }
    else if(C == 1869500027)
    {
        for(u32 i = ~-N; i;)
        {
            v[--i] = (u64(v[-~i]) * A + B) % 1869500027;
        }
    }
    else
    {
        for(u32 i = ~-N; i;)
        {
            u64 j = u64(v[-~--i]) * A + B;

            v[i] = C > j ? j : j - j / C * C;
        }
    }

    _index[0] = ~0;

    for(u32 i = N; i; ++frequence[v[--i] & 255]);

    for(u32 i = 0; i != 256; _index[++i] = _index[~-i] + frequence[~-i]);

    memset(frequence, 0, 1024);

    for(u32 i = 0; i != N; ++i) w[++_index[v[i] & 255]] = v[i], ++frequence[v[i] >> 8 & 255];


    _index[0] = ~0;

    for(u32 i = 0; i != 256; _index[++i] = _index[~-i] + frequence[~-i]);

    memset(frequence, 0, 1024);

    for(u32 i = 0; i != N; ++i) v[++_index[w[i] >> 8 & 255]] = w[i], ++frequence[w[i] >> 8 & 255];;


    _index[0] = ~0;

    for(u32 i = 0; i != 256; _index[++i] = _index[~-i] + frequence[~-i]);

    memset(frequence, 0, 1024);

    for(u32 i = 0; i != N; ++i) w[++_index[v[i] >> 16 & 255]] = v[i], ++frequence[v[i] >> 8 & 255];


    _index[0] = ~0;

    for(u32 i = 0; i != 256; _index[++i] = _index[~-i] + frequence[~-i]);

    for(u32 i = 0; i != N; ++i) v[++_index[w[i] >> 24 & 255]] = w[i];


    for(u32 i = N - 1 - (N - 1) % 10 + 10; i; i = i - 10, write(v[i]));

    std :: ofstream("radixsort.out").write(outBuffer + p, 10999999 - p);

    return 0;
}