HDU 5592 ZYB's Premutation

来源：互联网发布：java精品课程网编辑：程序博客网时间：2024/06/04 18:52

题意是给你n个位置的值，每个值是由之前的逆序数和加上当前的逆序数的总和。然后这个逆序数序列需要由你自己构建，由1~n的排列组成。
我们会发现首先最后一个位置可以直接确定，设它为a[i]，a[i]=当前的位置值减去之前的位置值总和。然后这个最后一个序列所填的值我们就可以确定的了，便就是n-a[n]，之后我们就需要再确定前一个位置所填的值，而要得到这个值的话，我们就需要从新序列里返回第n-a[i]小的元素的是哪一个，注意我们的这个1~n的序列在最开始把最后一个位置填的值确定后，新的序列要把这个填好的值从1~n里去掉，所以新的序列大小为–n，然后返回这个新序列里第k小元素的便是当前位置所填的值，之后再慢慢往前重复之前的操作。可以通过树状数组来进行操作，不会的可以看看这个链接:http://www.cnblogs.com/zgmf_x20a/archive/2008/11/15/1334109.html。
下面就附上代码

#include<stdio.h>#include<iostream>#include<string.h>#include<algorithm>using namespace std;const int maxn=1<<20;int c[maxn],a[50010],b[50010];int lowbit(int x){    return x&-x;}void insert(int x,int t){       while(x<maxn){          c[x]+=t;          x+=lowbit(x);           }}int find(int k){    int cnt=0,ans=0;    for(int i=20;i>=0;i--){        ans+=(1<<i);        if(ans>=maxn || cnt+c[ans]>=k)ans-=(1<<i);        else cnt+=c[ans];    }    return ans+1;}int main(){    int n,i,j,T;    scanf("%d",&T);    while(T--)    {        a[0]=0;        scanf("%d",&n);        int m=n,sum=0;        for(i=1;i<=n;i++)        {            insert(i,1);            scanf("%d",&a[i]);            a[i]-=sum;            sum+=a[i];        }        for(i=n;i>=1;i--)        {            b[i]=find(n-a[i]);            insert(b[i],-1);            --n;        }        for(i=1;i<m;i++)        {            printf("%d ",b[i]);        }        printf("%d\n",b[i]);    }}

0 0