Cod sursa(job #1419643)

Utilizator smaraldaSmaranda Dinu smaralda Data 16 aprilie 2015 00:44:27
Problema Curcubeu Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.04 kb
#include<stdio.h>
#include<algorithm>
using namespace std;

const int NMAX = 1e6 + 5;

int color[NMAX], nxt[NMAX], a[NMAX], b[NMAX], c[NMAX];

int getNext (int node) {
    if(nxt[node] == node)
        return node;
    return nxt[node] = getNext(nxt[node]);
}

int main() {
    freopen("curcubeu.in", "r", stdin);
    freopen("curcubeu.out", "w", stdout);
    int n, i, newLeft, left;

    scanf("%d%d%d%d", &n, &a[1], &b[1], &c[1]);
    
    for(i = 2; i <= n; ++ i) {
        nxt[i] = i;
        a[i] = (a[i - 1] * i) % n;
        b[i] = (b[i - 1] * i) % n;
        c[i] = (c[i - 1] * i) % n;
    }
    nxt[1] = 1;

    for(i = n - 1; i >= 1; -- i) {
        if(a[i] > b[i])
            swap(a[i], b[i]);

        left = getNext(a[i]);
        while(left <= b[i]) {
            color[left] = c[i];
            
            newLeft = getNext(left);

            nxt[left] = nxt[b[i]];
            left = newLeft;
            if(left == newLeft)
                ++ left;
        }
    }

    for(i = 1; i < n; ++ i)
        printf("%d\n", color[i]);
    return 0;
}