CSUOJ 1555 Inversion Sequence 线段树

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
#define INF 0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

typedef long long LL;

const int maxn=10005;

int sum[maxn<<2];
int pos,ok;
int n;
int k;
int ans[maxn];

void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=1;
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

void query(int t,int l,int r,int rt)
{
    if(t>sum[rt])
    {
        ok=0;
        return ;
    }
    if(l==r)
    {
        pos=l;
        sum[rt]=0;
        return ;
    }
    int m=(l+r)>>1;
    if(t<=sum[rt<<1])
        query(t,lson);
    else
        query(t-sum[rt<<1],rson);

        pushup(rt);
}

int main()
{
    //freopen("input.txt","r",stdin);
    while(~scanf("%d",&n))
    {
        ok=1;
        build(1,n,1);
        for(int i=1; i<=n; i++)
            {
                scanf("%d",&k);
                if(!ok)
                    continue;
                query(k+1,1,n,1);
                ans[pos]=i;
            }

        if(!ok)
        {
            printf("No solution\n");
            continue;
        }
        printf("%d",ans[1]);
        for(int i=2;i<=n;i++)
            printf(" %d",ans[i]);
        printf("\n");

    }
}