Cod sursa(job #1992356)

Utilizator Mihai_PredaPreda Mihai Dragos Mihai_Preda Data 20 iunie 2017 11:40:41
Problema Radix Sort Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.11 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

const int nMax = 10000008;
const int base = 4096;

inline int GetDigit(int x, int d)
{
    return (x / d) % base;
}

void count_sort(vector<int> &v, int digit)
{
    int n = v.size();
    vector<int> temp(n);
    vector<int> cnt(base, 0);
    for(auto x:v)
        cnt[GetDigit(x, digit)]++;
    for(int i = 1; i < base; ++i)
        cnt[i] = cnt[i-1] + cnt[i];
    for(int i = n-1; i >= 0; --i)
        temp[--cnt[GetDigit(v[i], digit)]] = v[i];
    v = temp;
}

void radix_sort(vector<int> &v)
{
    long long mx = *max_element(v.begin(), v.end());

    for(long long i = 1; mx / i > 0; i *= base)
        count_sort(v, i);
}

int main()
{
    int n, a, b, c;
    ifstream in("radixsort.in");
    in >> n >> a >> b >> c;
    in.close();
    vector<int> v(n);
    v[0] = b;
    for(int i = 1; i < n; ++i)
        v[i] = (1LL * a * v[i-1] + b) % c;
    radix_sort(v);

    ofstream out("radixsort.out");
    for(int i = 0; i < n; i += 10)
        out << v[i] << " ";
    return 0;
}