Cod sursa(job #2627068)

Utilizator euyoTukanul euyo Data 9 iunie 2020 16:36:41
Problema Curcubeu Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.05 kb
#include <fstream>
#define MAXQ 1000000

using namespace std;

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

int urm[MAXQ + 1];
int sol[MAXQ + 1];

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, rind;
  
  fin >> n >> a >> b >> c;
  for ( i = n - 1; i >= 1; --i ) {
	q[i].st = (a < b ? a : b);
	q[i].dr = (a + b) - q[i].st;
	q[i].col = c;
	urm[i] = i;
	a = ((long long)a * (n - i + 1)) % n;
	b = ((long long)b * (n - i + 1)) % n;
	c = ((long long)c * (n - i + 1)) % n;
  }
  for ( i = 1; i <= n - 1; ++i ) {
	ind = q[i].st;
	while ( ind <= q[i].dr ) {
	  rind = findRoot( ind );
	  if ( ind != q[i].st ) {
	    join( ind - 1, ind );
	  }
	  if ( sol[ind] == 0 ) {
		sol[ind] = q[i].col;
	  }
	  ind = rind + 1;
	}
  }
  for ( i = 1; i <= n - 1; ++i ) {
	fout << sol[i] << "\n";
  }
  fin.close();
  fout.close();
  return 0;
}