Cod sursa(job #2617811)

Utilizator srazvan100@gmail.comRazvan Alexandru Sandu [email protected] Data 22 mai 2020 22:38:01
Problema Radix Sort Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include <bits/stdc++.h>
using namespace std;

#define vi vector<int>

template <typename T>
ostream &operator<<(ostream &os, const vector<T> &v) {
    for (int i = 0; i < v.size(); ++i) {
        os << v[i];
        if (i != v.size() - 1) os << " ";
    }
    return os;
}

int getMax(vector<int> arr) {
    return *max_element(arr.begin(), arr.end());
}

void countSort(vi &arr, int e) {
    int n = arr.size();
    vi output(n);
    vi count(10);

    for (int i = 0; i < n; i++)
        count[(arr[i] / e) % 10]++;

    for (int i = 1; i < 10; i++)
        count[i] += count[i - 1];

    for (int i = n - 1; i >= 0; i--) {
        output[count[(arr[i] / e) % 10] - 1] = arr[i];
        count[(arr[i] / e) % 10]--;
    }

    for (int i = 0; i < n; i++)
        arr[i] = output[i];
}

void radix(vi &arr) {
    for (int e = 1; getMax(arr) / e > 0; e *= 10)
        countSort(arr, e);
}

int main() {
    ifstream fin("cautbin.in", ios::in);
    ofstream fout("cautbin.out", ios::out);

    int n, a, b, c;
    fin >> n >> a >> b >> c;
    vector<int> arr(n);
    arr[0] = b;
    for (int i = 1; i < n; ++i)
        arr[i] = (a * arr[i - 1] + b) % c;

    radix(arr);
    cout<<arr;

    return 0;
}