Cod sursa(job #1209942)

Utilizator daniel.amarieiDaniel Amariei daniel.amariei Data 18 iulie 2014 21:38:45
Problema Radix Sort Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <stdio.h>
#include <cstring>
#define MAX (10000000 + 1)
#define STEP 10
#define MASK_B1 255
#define MASK_B2 (MASK_B1 << 8)
#define MASK_B3 (MASK_B2 << 8)
#define MASK_B4 (MASK_B3 << 8)

int N, A, B, C;
int v[MAX], b[MAX], f[255];


void sort_after_byte(int * v, int * b, int MASK)
{
    for (int i = 1; i <= N; ++i)
    {
        ++f[v[i] & MASK];      
    }
    
    for (int i = 1; i <= 255; ++i)
    {
        f[i] += f[i-1];
    }
    
    for (int i = N; i >= 1; --i)
    {
        b[f[v[i] & MASK]] = v[i];
        --f[v[i] & MASK];
    }
}

int main()
{
    freopen("radixsort.in", "r", stdin);
    freopen("radixsort.out", "w", stdout);
    
    scanf("%d %d %d %d", &N, &A, &B, &C);
    
    // Generam numerele
    v[1] = B;
    for (int i = 2; i <= N; ++i)
    {
        v[i] = (A * v[i-1] + B) % C;
    }

    // Sortare
    sort_after_byte(v, b, MASK_B1);
    memset(f, 0, sizeof(f));

    sort_after_byte(b, v, MASK_B2);
    memset(f, 0, sizeof(f));

    sort_after_byte(v, b, MASK_B3);
    memset(f, 0, sizeof(f));

    sort_after_byte(b, v, MASK_B4);
    
    // Afisare rezultat
    for (int i = 1; i <= N; i+=STEP)
    {
        printf("%d ", v[i]);
    }

    return 0;
}