Cod sursa(job #2432240)

Utilizator capmareAlexCapmare Alex capmareAlex Data 22 iunie 2019 17:56:17
Problema Farfurii Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.37 kb
#include <bits/stdc++.h>
#define ll long long
#define NMAX 100001
using namespace std;
ifstream fin("farfurii.in");
ofstream fout("farfurii.out");
int n;
ll k;
vector< int > sol;
int aint[NMAX*4];
int query( int nod, int st, int dr, int l , int r)
{
    if( st >= l && dr<= r) return aint[nod];
    int mid = (st + dr)/2, ans = 0;
    if( mid >= l)
      ans = query(nod*2, st, mid, l, r);
    if( mid + 1 <= r)
        ans += query( nod * 2 + 1, mid+1, dr, l, r);
    return ans;
}
void update (int nod , int st, int dr, int pos, int val)
{
    if( st==dr)
        aint[nod] += val;
    else
    {
        int mid = (st+dr)/2;
        if(mid >= pos)
            update(nod*2, st, mid, pos, val);
        else update(nod*2+1, mid+1, dr, pos, val);
        aint[nod] = aint[nod*2]+aint[nod*2+1];
    }
}
int main()
{
    fin>>n>>k;
    for(int i=1;i<=n;++i)
    {
       int rms = 1ll*(n-i)*(n-i-1)/2;
       if(rms>k)
       {
           fout<<i<<" ";
           continue;
       }
       else if(rms==k)
       {
           fout<<i<<" ";
           for( int j=n; j >i; --j)
               fout<<j<<" ";
           break;
       }
       else
       {
           int nr= k- rms + 1;
           fout<<i-1 + nr<<" ";
           for(int j=n; j >=i; --j){if(i-1+nr==j)continue;cout<< j <<" ";}
           break;
       }
    }
    return 0;
}