Cod sursa(job #2196618)

Utilizator Lazar_LaurentiuLazar Laurentiu Lazar_Laurentiu Data 19 aprilie 2018 21:04:02
Problema Semne Scor 65
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.66 kb
#include <iostream>
#include <fstream>
#include <random>
#define MAX 50010

using namespace std;

int n,s,nrp,nr,st,dr,mij,sf;
int a[MAX];
bool sm[MAX];
int aib[MAX];
int nsb(int nr){return nr^(nr&(nr-1));}
void update(int nr,int incr){for(;nr<MAX;nr+=nsb(nr))aib[nr]+=incr;}
int query(int nr){
  int ansa=0;
  for(;nr;nr-=nsb(nr))ansa+=aib[nr];
  return ansa;
}

int main()
{
    ifstream f ("semne.in");
    ofstream g ("semne.out");
    f>>n>>sf; for(int i=1;i<=n;i++)f>>a[i],s+=a[i],sm[i]=1,update(i,1);
    nrp=n;
    default_random_engine generator;
    uniform_int_distribution<int> distribution(0,n-1);  //
    while(s!=sf){
      nr=distribution(generator);
      //cout<<nr<<'\n';
      if(s<sf){
        nr=nr%(n-nrp)+1;
        st=0,dr=n-1;
//        cout<<"\ns<sf\n";
//        for(int i=1;i<=n;i++)cout<<((sm[i])?"+ ":"- ");
//        cout<<'\n'<<nr<<" ";
        while(st<dr){
          mij=(st+dr+1)/2;
          if(mij-query(mij)<nr)st=mij;
          else dr=mij-1;
        }
        st++;
//        cout<<st<<'\n';
        s=s+2*a[st];sm[st]=1;nrp++;
        update(st,1);
      } else {
        nr=nr%nrp+1;
        st=0,dr=n-1;
//        cout<<"\ns>sf\n";
//        for(int i=1;i<=n;i++)cout<<((sm[i])?"+ ":"- ");
//        cout<<'\n'<<nr<<" ";
        while(st<dr){
          mij=(st+dr+1)/2;
          if(query(mij)<nr)st=mij;
          else dr=mij-1;
        }
        st++;
//        cout<<st<<'\n';
        s=s-2*a[st];sm[st]=0;nrp--;
        update(st,-1);
      }
    }
    for(int i=1;i<=n;i++)
      if(sm[i]==0)g<<'-';
      else g<<'+';
    g<<'\n';
    f.close ();
    g.close ();
    return 0;
}