Cod sursa(job #3331334)

Utilizator boboc132Boboc Teodor boboc132 Data 26 decembrie 2025 18:26:00
Problema Curcubeu Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.85 kb
#include <iostream>
#include <fstream>
#include <algorithm>

using namespace std;

ifstream in("curcubeu.in");
ofstream out("curcubeu.out");

const int MAXN = 1000005;

int L[MAXN], R[MAXN], C[MAXN];
int Next[MAXN], solutie[MAXN];
int N;

// DSU Iterativ - Blindat
int find_next(int pos) {
    int root = pos;
    while (Next[root] != root) root = Next[root];
    int aux = pos;
    while (Next[aux] != root) {
        int next_node = Next[aux];
        Next[aux] = root;
        aux = next_node;
    }
    return root;
}

int main() {
    ios_base::sync_with_stdio(false);
    in.tie(NULL);

    long long Ar, Br, Cr;
    if (!(in >> N >> Ar >> Br >> Cr)) return 0;

    // ETAPA 1: Generarea (Exact ca in enunt)
    // A_1, B_1, C_1 sunt cele citite. A_i se bazeaza pe A_{i-1}
    for (int i = 1; i < N; i++) {
        L[i] = (int)(min(Ar, Br) + 1);
        R[i] = (int)(max(Ar, Br) + 1);
        C[i] = (int)Cr;

        // Pregatim valorile pentru pasul i + 1
        // Folosim 1LL pentru a forca inmultirea pe long long
        Ar = (1LL * Ar * i) % N;
        Br = (1LL * Br * i) % N;
        Cr = (1LL * Cr * i) % N;
    }

    // ETAPA 2: Initializare DSU
    for (int i = 1; i <= N + 1; i++) Next[i] = i;

    // ETAPA 3: Procesare INVERSA
    // Ultima pensula aplicata (N-1) e prima care coloreaza definitiv o celula
    for (int i = N - 1; i >= 1; i--) {
        int st = L[i];
        int dr = R[i];
        
        for (int j = find_next(st); j <= dr; j = find_next(j)) {
            // Daca j == N, nu ne intereseaza pentru afisare, dar trebuie "sters" din DSU
            if (j < N) solutie[j] = C[i];
            
            Next[j] = find_next(j + 1);
        }
    }

    // ETAPA 4: Afisare (celulele 1 ... N-1)
    for (int i = 1; i < N; i++) {
        out << solutie[i] << "\n";
    }

    return 0;
}