Cod sursa(job #2018534)

Utilizator cipri321Marin Ciprian cipri321 Data 5 septembrie 2017 12:27:20
Problema Order Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.19 kb
#include <fstream>
#define DIM 1<<16
using namespace std;
ifstream fi("order.in");
ofstream fo("order.out");
int n;
int M[DIM];
void init(int nod,int l,int r)
{
    if(l>r)
        return;
    if(l==r)
        M[nod]=1;
    else
    {
        init(2*nod,l,(l+r)/2);
        init(2*nod+1,(l+r)/2+1,r);
        M[nod]=M[2*nod]+M[2*nod+1];
    }
}
void update(int nod,int l,int r,int poz, int val)
{
    if(l>r)
        return ;
    if(l==r && l!=poz)
        return ;
    if(l==r)
        M[nod]=val;
    else
    {
        update(2*nod,l,(l+r)/2,poz,val);
        update(2*nod+1,(l+r)/2+1,r,poz,val);
        M[nod]=M[2*nod]+M[2*nod+1];
    }
}
int query(int nod,int l,int r,int x)
{
    if(l>r)
        return -1;
    if(l==r)
        return l;
    if(x-M[2*nod]<=0)
        return query(2*nod,l,(l+r)/2,x);
    else
        return query(2*nod+1,(l+r)/2+1,r,x-M[2*nod]);

}
int main()
{
    fi>>n;
    init(1,1,n);
    int pas=2;
    for(int i=1;i<=n;i++)
    {
        int poz=query(1,1,n,pas);
        fo<<poz<<" ";
        update(1,1,n,poz,0);
        if(M[1]!=0)
            pas=(pas+i-1)%M[1]+1;
    }
    fi.close();
    fo.close();
    return 0;
}