Cod sursa(job #2301066)

Utilizator AlexPop28Pop Alex-Nicolae AlexPop28 Data 12 decembrie 2018 16:51:36
Problema Radix Sort Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.38 kb
#include <bits/stdc++.h>
#define all(cont) cont.begin(), cont.end()
#define pb push_back
#define fi first
#define se second
#define DEBUG(x) cerr << (#x) << ": " << (x) << '\n'

using namespace std;

typedef pair <int, int> pii;
typedef vector <int> vi;
typedef long long ll;
typedef unsigned long long ull;

ifstream f ("radixsort.in");
ofstream g ("radixsort.out");

const int SHIFT = 8;
const int NMAX = 1e7 + 5;

ll n, a, b, c;
int v[NMAX], aux[NMAX];
int cnt[(1 << SHIFT) + 5], idx[(1 << SHIFT) + 5];

int main() {
  ios::sync_with_stdio(0);
  cin.tie(0);
#ifdef LOCAL_DEFINE
  freopen (".in", "r", stdin);
#endif

  f >> n >> a >> b >> c;
  v[1] = b;

  for (int i = 2; i <= n; ++i) {
    v[i] = (1LL * a * v[i - 1] + b) % c;
  }

  int and_val = (1 << SHIFT) - 1;

  for (int bits = 0; bits < 4; ++bits) {
    for (int i = 1; i <= n; ++i) {
      ++cnt[((v[i] >> (bits * SHIFT)) & and_val)];
    }

    idx[0] = 1;
    for (int i = 1; i <= (1 << SHIFT); ++i) {
      idx[i] = idx[i - 1] + cnt[i - 1];
    }

    for (int i = 1; i <= n; ++i) {
      aux[idx[((v[i] >> (bits * SHIFT)) & and_val)]++] = v[i];
    }

    for (int i = 1; i <= n; ++i) {
      cnt[((v[i] >> (bits * SHIFT)) & and_val)] = 0;
      v[i] = aux[i];
    }
  }

  for (int i = 1; i <= n; i += 10) {
    g << v[i] << ' ';
  }

  f.close();
  g.close();
  return 0;
}