Cod sursa(job #2443056)

Utilizator iulianarsenoiuArsenoiu Iulian iulianarsenoiu Data 26 iulie 2019 12:26:16
Problema Farfurii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <fstream>

using namespace std;
ifstream f("farfurii.in");
ofstream g("farfurii.out");
long long n,k,v[100005],i,d,aib[100005];
long long ub(long long x)
{
    return (x&(-x));
}
void Update(long long x, long long val)
{
    for(long long i=x;i<=n;i+=ub(i))
        aib[i]+=val;
}
long long query(long long x)
{
    long long s=0;
    for(long long i=x;i>=1;i-=ub(i))
        s+=aib[i];
    return s;
}
long long cautbin(long long x)
{
    long long st=1, dr=n,mij=0;
    long long s=0, poz=0;
    while(st<=dr)
    {
        mij=(st+dr)/2;
        s=query(mij);
        if(x<=s)
        {
            poz=mij;
            dr=mij-1;
        }
        else
            st=mij+1;
    }
    return poz;
}
int main()
{
    f>>n>>k;
    for(i=1;i<=n;i++)
        Update(i,1);
    for(i=1;i<=n;i++)
    {
        if((n-i)*(n-i-1)/2>=k)
            d=0;
        else
            d=k-(n-i)*(n-i-1)/2;
        v[i]=cautbin(d+1);
        Update(v[i],-1);
        k-=d;
    }
    for(i=1;i<=n;i++)
        g<<v[i]<<' ';
    return 0;
}