Cod sursa(job #3146572)

Utilizator Alex_tz307Lorintz Alexandru Alex_tz307 Data 21 august 2023 18:43:38
Problema Radix Sort Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.98 kb
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fstream>

using namespace std;

class OutParser {
private:
    FILE *fout;
    char *buff;
    int sp;
    void write_ch(char ch) {
        if (sp == 50000) {
            fwrite(buff, 1, 50000, fout);
            sp = 0;
            buff[sp++] = ch;
        } else {
            buff[sp++] = ch;
        }
    }
public:
    OutParser(const char* name) {
        fout = fopen(name, "w");
        buff = new char[50000]();
        sp = 0;
    }
    ~OutParser() {
        fwrite(buff, 1, sp, fout);
        fclose(fout);
    }
    OutParser& operator << (int vu32) {
        if (vu32 <= 9) {
            write_ch(vu32 + '0');
        } else {
            (*this) << (vu32 / 10);
            write_ch(vu32 % 10 + '0');
        }
        return *this;
    }
    OutParser& operator << (char ch) {
        write_ch(ch);
        return *this;
    }
    OutParser& operator << (const char *ch) {
        while (*ch) {
            write_ch(*ch);
            ++ch;
        }
        return *this;
    }
};

ifstream fin("radixsort.in");
OutParser fout("radixsort.out");

#define get_byte(x) ((x >> (byte * 16)) & RADIX)

const int kN = 1e7;
const int RADIX = 65535;
const int RADIX_SIZE = 16;
int n, numbers[kN], temp[kN];

void CountSort (int A[], int B[], int byte) {
  int counter[1 << RADIX_SIZE], index[1 << RADIX_SIZE];
  memset(counter, 0, sizeof(counter));

  for (int i = 0; i < n; ++i) {
    counter[get_byte(A[i])] += 1;
  }

  index[0] = 0;
  for (int i = 1; i < (1 << RADIX_SIZE); ++i) {
    index[i] = index[i - 1] + counter[i - 1];
  }

  for(int i = 0; i < n; ++i) {
    B[index[get_byte(A[i])]++] = A[i];
  }
}

int main () {
  int a, b, c;
  fin >> n >> a >> b >> c;

  numbers[0] = b % c;
  for (int i = 1; i < n; ++i) {
    numbers[i] = (1LL * a * numbers[i - 1] + b) % c;
  }
    
  CountSort (numbers, temp, 0);
  CountSort (temp, numbers, 1);

  for (int i = 0; i < n; i += 10) {
    fout << numbers[i] << ' ';
  }

  fin.close();
  return 0;
}