Cod sursa(job #3169229)

Utilizator tudor_costinCostin Tudor tudor_costin Data 14 noiembrie 2023 14:09:37
Problema Order Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.11 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("order.in");
ofstream fout("order.out");
const int Nmax=30005;
int n;
int aib[Nmax];
int lsb(int x)
{
    return x&(-x);
}
void update(int poz,int val)
{
    for(int i=poz;i<=n;i+=lsb(i))
    {
        aib[i]+=val;
    }
}
int query(int poz)
{
    int sol=0;
    for(int i=poz;i>=1;i-=lsb(i))
    {
        sol+=aib[i];
    }
    return sol;
}
int bfind(int a)
{
    int st=1,dr=n;
    while(st<=dr)
    {
        int mid=(st+dr)/2;
        if(query(mid)<a)
        {
            st=mid+1;
        }
        else
        {
            dr=mid-1;
        }
    }
    return st;
}
int main()
{
    fin>>n;
    for(int i=1;i<=n;i++)
    {
        aib[i]=lsb(i);
    }
    if(n<2)
    {
        fout<<1<<' ';
        return 0;
    }
    int prev=1;
    for(int i=1;i<=n;i++)
    {
        int scoase=prev-query(prev);
        int nr=(prev-scoase+i)%(n-i+1);
        if(nr==0) nr=n-i+1;
        int poz=bfind(nr);
        fout<<poz<<' ';
        update(poz,-1);
        prev=poz;
    }
    return 0;
}