HDU 5775 Bubble Sort(BIT)

来源:互联网 发布:linux echo 颜色 编辑:程序博客网 时间:2024/05/16 09:00

Description
P序列是一个1~n的重排,现在对P序列进行如下操作,问1~n每个数在操作过程中能到达的最右边位置与最左边位置之差是多少
for(int i=1;i<=N;++i)
for(int j=N,t;j>i;—j)
if(P[j-1] > P[j])
t=P[j],P[j]=P[j-1],P[j-1]=t;
Input
第一行一整数T表示用例组数,每组用例首先输入一整数n表示序列长度,之后n个整数表示P序列(T<=20,1<=n<=10^5)
Output
Sample Input
2
3
3 1 2
3
1 2 3
Sample Output
Case #1: 1 1 2
Case #2: 0 0 0
Solution
ans[a[i]]=max(a[i],i+i右边比a[i]小的数的数量)-min(a[i],i)
从后往前把该序列的数一个个插到树状数组中即可求出每个数右边比其小的数的个数,时间复杂度O(nlogn)
Code

#include<cstdio>#include<iostream>#include<cstring>#include<algorithm>using namespace std;#define maxn 111111struct BIT {    #define lowbit(x) (x&(-x))    int b[maxn];    void init()    {        memset(b,0,sizeof(b));    }    void update(int x,int v)    {        while(x<maxn)        {            b[x]+=v;            x+=lowbit(x);        }    }    int query(int x)    {        int ans=0;        while(x)        {            ans+=b[x];            x-=lowbit(x);        }        return ans;    }};int T,n,a[maxn],Case=1,ans[maxn];BIT bit;int main(){    scanf("%d",&T);    while(T--)    {        scanf("%d",&n);        for(int i=1;i<=n;i++)scanf("%d",&a[i]);        bit.init();        for(int i=n;i>=1;i--)        {            int temp=bit.query(a[i]);            ans[a[i]]=max(a[i],i+temp)-min(a[i],i);            bit.update(a[i],1);        }        printf("Case #%d:",Case++);        for(int i=1;i<=n;i++)printf(" %d",ans[i]);        printf("\n");    }    return 0;}
0 0
原创粉丝点击