Cod sursa(job #2443478)

Utilizator filiptudose2007Tudose Filip filiptudose2007 Data 28 iulie 2019 11:53:16
Problema Farfurii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <bits/stdc++.h>
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 cb(long long x)
{
    long long st=1,dr=n,mij=0,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]=cb(d+1);
        Update(v[i],-1);
        k-=d;
    }
    for(i=1;i<=n;++i)g<<v[i]<<' ';
    return 0;
}