Cod sursa(job #2627040)

Utilizator euyoTukanul euyo Data 9 iunie 2020 14:17:21
Problema Curcubeu Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <fstream>
#define MAXQ 1000000

using namespace std;

struct update {
  int st, dr, col;
} q[MAXQ];

int urm[MAXQ];
int sol[MAXQ];

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

int findRoot( int x ) {
  if ( urm[x] == x ) {
    return x;
  }
  return urm[x] = findRoot( urm[x] );
}
void join( int x, int y ) {
  urm[x] = urm[findRoot( x )] = findRoot( y );
}

int main() {
  int n, i, a, b, c, ind;
  
  fin >> n >> a >> b >> c;
  for ( i = n - 2; i >= 0; --i ) {
	q[i].st = a < b ? a : b - 1;
	q[i].dr = a + b - q[i].st - 1;
	q[i].col = c;
	urm[i] = i;
	a = (a * (i + 1)) % n;
	b = (b * (i + 1)) % n;
	c = (c * (i + 1)) % n;
  }
  for ( i = 0; i < n - 1; ++i ) {
	ind = q[i].st;
	while ( ind < q[i].dr ) {
	  if ( sol[ind] == 0 && sol[ind + 1] == 0 ) {
        join( ind, ind + 1 );
	    sol[ind] = q[i].col;
		++ind;
	  } else {
        ind = findRoot( ind ) + 1;
	  }
	}
  }
  for ( i = 0; i < n - 1; ++i ) {
	fout << sol[i] << "\n";
  }
  fin.close();
  fout.close();
  return 0;
}