Cod sursa(job #2126536)

Utilizator MaligMamaliga cu smantana Malig Data 9 februarie 2018 18:37:49
Problema Radix Sort Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.38 kb
#include <iostream>
#include <fstream>
#include <cstring>
#include <vector>
#include <algorithm>
#include <queue>

using namespace std;
ifstream in("radixsort.in");
ofstream out("radixsort.out");

#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
const int NMax = 1e7 + 5;
const int inf = 1e9 + 5;
const int Radix = 0b11111111;
const int bits = 8;
const int nrBytes = 4;

int v1[NMax],v2[NMax];

void radixSort(int*,int*,int,int);

int main() {
    ll N,A,B,C;
    in>>N>>A>>B>>C;

    v1[1] = B;
    A %= C;
    B %= C;
    for (int i=2;i <= N;++i) {
        v1[i] = (A * v1[i-1] + B) % C;
    }

    for (int i=0;i < nrBytes;++i) {
        if (i % 2 == 0) {
            radixSort(v1,v2,i,N);
        }
        else {
            radixSort(v2,v1,i,N);
        }
    }

    for (int i=1;i <= N;i += 10) {
        out<<v1[i]<<' ';
    }

    in.close();out.close();
    return 0;
}

void radixSort(int* one,int* two,int byte,int N) {
    int occ[Radix + 5] = {};

    for (int i=1;i <= N;++i) {
        ++occ[ (one[i]>>(byte * bits)) & Radix ];
    }

    int total = 0;
    for (int i=0;i <= Radix;++i) {
        int old = occ[i];
        occ[i] = total + 1;
        total += old;
    }

    for (int i=1;i <= N;++i) {
        two[ occ[ (one[i]>>(byte * bits)) & Radix ]++ ] = one[i];
    }
}