Cod sursa(job #2400469)

Utilizator onipreponiprep oniprep Data 8 aprilie 2019 19:39:33
Problema Farfurii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.76 kb
#include <bits/stdc++.h>
using namespace std;
long long n,i,arb[300010],poz,k;
ofstream fout("farfurii.out");
ifstream fin("farfurii.in");
void update(int node,int l,int r,long long lib)
{
    if(l==r)
    {
        arb[node]=1;
        fout<<l<<" ";
        return ;
    }
    int mid=(l+r)/2;
    if(lib<=(mid-l+1)-arb[node*2])
        update(node*2,l,mid,lib);
    else
        update(node*2+1,mid+1,r,lib-((mid-l+1)-arb[node*2]));
    arb[node]=arb[node*2]+arb[node*2+1];
}
int main()
{
    fin>>n>>k;
       for(i=1; i<=n; ++i)
           if(k<=(n-i)*(n-i-1)/2)
               update(1,1,n,1);
        else
        {
            poz=k-(n-i)*(n-i-1)/2+1;
            k=(n-i)*(n-i-1)/2;
            update(1,1,n,poz);
        }
    return 0;
}