HDU5592(線段樹)

題目大意:給你一個序列,表示在這個位置以及之前,有p[i]個逆序對,求此序列。

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<iostream>
#include<set>
#include<map>
#include<queue>
#include<stack>
using namespace std;
#define lson o<<1,l,m
#define rson o<<1|1,m+1,r
const int maxn=55555;
int a[maxn<<2],ans[maxn<<2];
int sum[maxn<<2];
void pushup(int o)
{
    sum[o]=sum[o<<1]+sum[o<<1|1];
}
void build(int o,int l,int r)
{
    if(l==r)
    {
        sum[o]=r-l+1;
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(o);
}
int query(int o,int l,int r,int k)
{
    if(l==r)
    {
        sum[o]=0;
        return l;
    }
    int res;
    int m=(l+r)>>1;
    if(sum[o<<1|1]>=k)
        res=query(rson,k);
    else
        res=query(lson,k-sum[o<<1|1]);
    pushup(o);
    return res;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        a[0]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        build(1,1,n);
        for(int i=n;i>=1;i--)
        {
            int x=a[i]-a[i-1];
            ans[i]=query(1,1,n,x+1);
        }
        for(int i=1;i<=n;i++)
        {
            printf("%d%c",ans[i],(i==n)?'\n':' ');
        }
    }
    return 0;
}


發佈了28 篇原創文章 · 獲贊 4 · 訪問量 5827
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章