2016 Multi-University Training Contest 4 Bubble Sort(树状数组模板)
Bubble Sort
题意:
给你一个1~n的排列,问冒泡排序过程中,数字i(1<=i<=n)所到达的最左位置与最右位置的差值的绝对值是多少
题解:
数字i多能到达的最左位置为min(s[i],i)
i为它的初始位置,s[i]为它的最终位置(因为最后是排好序,这个数是多少,就排在哪个位置,故为s[i])
那最右位置呢?
就是判断数i初始时,右边有多少个数比i小,这个就能用树状数组来求解了
循环从右到左,对于数s[i],我们只需判断它右边1~s[i]-1中有几个数即可
树状数组是从1开始,所以输入尽量也从1开始
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const ll LINF=0x3f3f3f3f3f3f3f3f;
#define PI(A) cout<<(A)<<endl
#define SI(N) cin>>(N)
#define SII(N,M) cin>>(N)>>(M)
#define cle(a,val) memset(a,(val),sizeof(a))
#define rep(i,b) for(int i=0;i<(b);i++)
#define Rep(i,a,b) for(int i=(a);i<=(b);i++)
#define reRep(i,a,b) for(int i=(a);i>=(b);i--)
#define dbg(x) cout <<#x<<" = "<<(x)<<endl
#define PIar(a,n) rep(i,n)cout<<a[i]<<" ";cout<<endl;
#define PIarr(a,n,m) rep(aa,n){rep(bb, m)cout<<a[aa][bb]<<" ";cout<<endl;}
const double EPS= 1e-9 ;
/* ///////////////////////// C o d i n g S p a c e ///////////////////////// */
const int MAXN= 100000+ 9 ;
int s[MAXN],N,l[MAXN],r[MAXN];
//树状数组 范围是[1,n]
int bit[MAXN];
//求前i项和
int SUM(int i)
{
int s=0;
while(i>0)
{
s+=bit[i];
i-=i&-i;
}
return s;
}
//bit[i]+x
void ADD(int i,int x)
{
while(i<=MAXN)
{
bit[i]+=x;
i+=i&-i;
}
}
int main()
{
int o;
SI(o);
Rep(T,1,o)
{
cle(bit,0);
int ma=0;
SI(N);
Rep(i,1,N) SI(s[i]);
Rep(i,1,N) l[s[i]]=min(s[i],i);
for (int i=N;i>0;i--)
{
r[s[i]]=i+SUM(s[i]-1);
ADD(s[i],1);
}
cout<<"Case #"<<T<<":";
Rep(i,1,N) cout<<" "<<abs(l[i]-r[i]);
cout<<endl;
}
return 0;
}