Cod sursa(job #2117404)

Utilizator TincaMateiTinca Matei TincaMatei Data 28 ianuarie 2018 20:39:18
Problema Schi Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.5 kb
#include <cstdio>

const int MAX_N = 30000;

int aib[ 1 + MAX_N ];
int locul[ MAX_N ];
int locPartial[ MAX_N ];
 
//determinam ultimul bit al lui n
static inline int lastBit( int x ) { return x & -x; }
 
//adaugam val la pozitia i si actualizam aib-ul
static inline void add( int i , int val , int n ) {
  while( i <= n ) {
    aib[ i ] += val;
    i += lastBit( i );
  }
}
 
//calculam suma de la 1 la i in aib
static inline int suma( int i ) {
  int s;
  s = 0;
  while( i > 0 ) {
    s += aib[ i ];
    i -= lastBit( i );
  }
  return s;
}
 
//calculeaza locul in clasamentul partial pt locul dat
inline int calculLoc( int loc ) { return loc - suma( loc - 1 ); }
 
//cauta pe ce loc se afla cel de pe locul dat in clasamentul partial
int cautare( int loc , int n ) {
  int min , max , mid , rez;
  min = 0;
  max = n;
  while( max - min > 1 ) {
    mid = ( min + max ) >> 1;
    rez = calculLoc( mid + 1 );
    if( rez <= loc )
      min = mid;
    else
      max = mid;
  }
  return min;
}
 
int main() {
  int n , i , loc ;
  FILE *fin = fopen("schi.in", "r");
  fscanf(fin, "%d", &n);
  for( i = 0 ; i < n ; i++ )
    fscanf(fin, "%d", &locPartial[i]);
  fclose(fin);

  //pt fiecare concurent caut locul lui in clasamentul real
  for( i = n - 1 ; i >= 0 ; i-- ) {
    loc = cautare( locPartial[ i ] , n );
    locul[ loc ] = i + 1;
    add( loc + 1 , 1 , n );
  }
 
  FILE *fout = fopen( "schi.out" , "w" );
  //scriu clasamentul
  for( i = 0 ; i < n ; i++ )
    fprintf( fout , "%d\n" , locul[ i ] );
  fclose( fout );
  return 0;
}