Cod sursa(job #1954671)

Utilizator the.manIon Man the.man Data 5 aprilie 2017 16:10:39
Problema Schi Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <fstream>
using namespace std;
ifstream fi("schi.in");
ofstream fo("schi.out");

int A[30001],v[30001],sol[30001],i,n;

void Update(int x,int v)
{
    for(int i=x; i<=n; i+= i & (-i))
        A[i]+=v;
}
int  Query(int x)
{
    int s=0;
    for(int i=x; i>=1; i-= i & (-i))
        s+=A[i];
    return s;
}

int Search(int p)
{
    int st = 1, dr = n, sol = n;
    while (st <= dr)
    {
        int mij = (st + dr) >> 1;
        if (Query(mij) >= p)
        {
            sol = mij;
            dr = mij - 1;
        }
        else
            st = mij + 1;
    }
    return sol;
}
int main()
{
 fi>>n;
 for(int i=1;i<=n;i++) fi>>v[i];
 for(int i=1;i<=n;i++) Update(i,1);

 for(i=n;i>=1;i--)
 {
    int poz=Search(v[i]);
    Update(poz,-1);
    sol[poz]=i;
 }

 for(int i=1;i<=n;i++) fo<<sol[i]<<" ";
    return 0;
}