Cod sursa(job #2635784)

Utilizator popoviciAna16Popovici Ana popoviciAna16 Data 15 iulie 2020 15:39:02
Problema Farfurii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.51 kb
#include <fstream>
using namespace std;

ifstream fin("farfurii.in");
ofstream fout("farfurii.out");

int a[400001], poz;

void constr(int nod, int st, int dr)
{
    if (st == dr)
        a[nod] = 1;
    else
    {
        int mij = (st+dr)>>1;
        constr(nod<<1, st, mij);
        constr(nod<<1|1, mij+1, dr);
        a[nod] = a[nod<<1] + a[nod<<1|1];
    }
}

void update(int nod, int st, int dr)
{
    if (st == dr)
        a[nod] = 0;
    else
    {
        int mij = (st+dr)>>1;
        if (poz <= mij)
            update(nod<<1, st, mij);
        else
            update(nod<<1|1, mij+1, dr);
        a[nod] = a[nod<<1] + a[nod<<1|1];
    }
}

int query(int n, int k) //imi returneaza pozitia celui de-al k-lea 1
{
    //fout << k << ' ';
    int nod = 1, st = 1, dr = n, mij;
    while (st != dr)
    {
        mij = (st+dr)>>1;
        if (k <= a[nod<<1])
        {
            dr = mij;
            nod = nod<<1;
        }
        else
        {
            k = k - a[nod<<1];
            st = mij+1;
            nod = nod<<1|1;
        }
    }
    return st;
}

int main()
{
    int n, i;
    long long k;
    fin >> n >> k;
    constr(1, 1, n);
    for (i = 1; i<=n; i++)
    {
        if (k <= (1ll*n-i)*(n-i-1)/2 && i < n)
        {
            fout << i << ' ';
            poz = i;
            update(1, 1, n);
        }
        else
        {
            poz = query(n, k - (1ll*n-i)*(n-i-1)/2+1);
            k = (1ll*n-i)*(n-i-1)/2;
            fout << poz << ' ';
            update(1, 1, n);
        }
    }
    return 0;
}