Cod sursa(job #3191773)

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

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

int n,a,b,c;
vector<int> tatici;
vector<int> colorat;
vector<int> leftSide;
vector<int> rightSide;
vector<int> color;
int getNext(int k) {
    int mem = k;
    while(tatici[k] != -1) {
        k = tatici[k];
    }
    while(tatici[mem] != k && tatici[mem] != -1) {
        const int aux = tatici[mem];
        tatici[mem] = k;
        mem = aux;
    }
    return k;
}

int main() {
    fin >> n;

    tatici.assign(MAX_N, -1);
    colorat.assign(n, 0);
    leftSide.resize(n);
    rightSide.resize(n);
    color.resize(n);

    fin >> leftSide[1] >> rightSide[1] >> color[1];
    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 = _left; k <= _right; k++) {
            colorat[k] = color[i];
            tatici[k] = getNext(k+1);
        }
    }
    for (int i = 1; i < n; i++) {
        fout << colorat[i] << '\n';
    }

    return 0;
}