Cod sursa(job #2691140)

Utilizator cristi_macoveiMacovei Cristian cristi_macovei Data 27 decembrie 2020 13:29:23
Problema Curcubeu Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <iostream>
#include <fstream>
#include <algorithm>

const int NMAX = 1e6;

int n;

int color[1 + NMAX];

int next[1 + NMAX];

int a[1 + NMAX], b[1 + NMAX], c[1 + NMAX];

int getNext(int i) {
  if (color[i] == 0)
    return i;

  next[i] = getNext(next[i + 1]);
  return next[i];
}

int main() {
  std::ifstream in("curcubeu.in");
  std::ofstream out("curcubeu.out");

  in >> n >> a[1] >> b[1] >> c[1];

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

    if (a[i] > b[i])
      std::swap(a[i], b[i]);
  }

  for (int i = 1; i <= n; ++i)
    next[i] = i;

  for (int q = n - 1; q >= 1; --q) {
    for (int i = a[q]; i <= b[q]; i = getNext(i + 1)) {
      if (color[i] == 0) {
        color[i] = c[q];
        next[i] = getNext(b[q] + 1);
      }
    }
  }

  for (int i = 1; i < n; ++i)
    out << color[i] << '\n';

  return 0;
}