Cod sursa(job #3191780)

Utilizator Manolea_Teodor_StefanManolea Teodor Stefan Manolea_Teodor_Stefan Data 10 ianuarie 2024 17:19:59
Problema Curcubeu Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <bits/stdc++.h>
#define MAX_N 1000005

using namespace std;
ifstream fin("curcubeu.in");
ofstream fout("curcubeu.out");

int n,a,b,c;
array<int, MAX_N> tatici;
int colorat[MAX_N];
array<int, MAX_N> leftSide;
array<int, MAX_N> rightSide;
array<int, MAX_N> color;
int getNext(int pos) {
    int mem = pos;
    while(tatici[mem] != pos) {
        pos = tatici[pos];
    }
    int res = pos;
    pos = mem;
    while(tatici[pos] != res) {
        mem = tatici[pos];
        tatici[pos] = res;
        pos = mem;
    }
    return res;
}

int main() {
    fin >> n;

    fin >> leftSide[1] >> rightSide[1] >> color[1];
    for(int i = 1; i < MAX_N; i++) {
        tatici[i] = i;
    }
    for(int i = 2; i < n; i++) {

        leftSide[i] = (1ll*leftSide[i-1]*i) % n;
        rightSide[i] = (1ll*rightSide[i-1]*i) % n;
        color[i] = (1ll*color[i-1]*i) % n;
    }

    for (int i = n-1; i > 0; i--) {

        const int _left = min(leftSide[i],rightSide[i]);
        const int _right = max(leftSide[i],rightSide[i]);
        for (int k = getNext(_left); k <= _right; k = tatici[k]) {
            colorat[k] = color[i];
            tatici[k] = getNext(k+1);
        }
    }
    for (int i = 1; i < n; i++) {
        fout << colorat[i] << '\n';
    }

    return 0;
}