BZOJ2141_排队_树状数组+分块
Description
排排坐，吃果果，生果甜嗦嗦，大家笑呵呵。你一个，我一个，大的分给你，小的留给我，吃完果果唱支歌，大家
乐和和。红星幼儿园的小朋友们排起了长长地队伍，准备吃果果。不过因为小朋友们的身高有所区别，排成的队伍
高低错乱，极不美观。设第i个小朋友的身高为hi，我们定义一个序列的杂乱程度为：满足ihj的(i,j)数量。幼儿
园阿姨每次会选出两个小朋友，交换他们的位置，请你帮忙计算出每次交换后，序列的杂乱程度。为方便幼儿园阿
姨统计，在未进行任何交换操作时，你也应该输出该序列的杂乱程度。
Input
第一行为一个正整数n，表示小朋友的数量；
第二行包含n个由空格分隔的正整数h1,h2,…,hn，依次表示初始队列中小朋友的身高；
第三行为一个正整数m，表示交换操作的次数；
以下m行每行包含两个正整数ai和bi，表示交换位置ai与位置bi的小朋友。
1≤m≤2*10^3，1≤n≤2*104，1≤hi≤109，ai≠bi，1≤ai,bi≤n。
Output
输出文件共m行，第i行一个正整数表示交换操作i结束后，序列的杂乱程度。
Sample Input
【样例输入】
3
130 150 140
2
2 3
1 3
3
130 150 140
2
2 3
1 3
Sample Output
1
0
3
【样例说明】
未进行任何操作时，(2,3)满足条件；
操作1结束后，序列为130 140 150，不存在满足ihj的(i,j)对；
操作2结束后，序列为150 140 130,(1,2)，(1,3)，(2,3)共3对满足条件的(i,j)
0
3
【样例说明】
未进行任何操作时，(2,3)满足条件；
操作1结束后，序列为130 140 150，不存在满足ihj的(i,j)对；
操作2结束后，序列为150 140 130,(1,2)，(1,3)，(2,3)共3对满足条件的(i,j)
每次交换两个数，求整个序列的逆序对。
将序列分成根号n块， 块内用树状数组维护每个数出现的个数。
修改L，R时发现对答案产生影响的只有L，R中间的这些数。
整块的树状数组求出现次数，零散的暴力找。
空间复杂度nsqrt(n) 时间复杂度nsqrt(n)
代码：
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
using namespace std;
int n,m,L[250],R[250],block,pos[20050],ans;
struct A {
int num,id,v;
}a[20050];
struct BIT {
int c[20050];
int inq(int x) {
int re=0;
for(;x;x-=x&(-x)) re+=c[x];
return re;
}
void fix(int x,int v) {
for(;x<=n;x+=x&(-x)) c[x]+=v;
}
}b[250],t;
bool cmp1(const A &x,const A &y) {return x.num<y.num;}
bool cmp2(const A &x,const A &y) {return x.id<y.id;}
int query(int l,int r,int x) {
int p=pos[l],q=pos[r],re=0,i;
if(p==q) {
for(i=l;i<=r;i++) {
if(a[i].v<=x) re++;
}
return re;
}
for(i=p+1;i<=q-1;i++) {
re+=b[i].inq(x);
}
for(i=l;i<=R[p];i++) {
if(a[i].v<=x) re++;
}
for(i=L[q];i<=r;i++) {
if(a[i].v<=x) re++;
}
return re;
}
int main() {
scanf("%d",&n);
int i,j,x,y;
for(i=1;i<=n;i++) scanf("%d",&a[i].num),a[i].id=i;
sort(a+1,a+n+1,cmp1); a[0].num=453453;
for(i=1,j=0;i<=n;i++) {if(a[i].num!=a[i-1].num) j++; a[i].v=j;}
sort(a+1,a+n+1,cmp2);
block=sqrt(n);
for(i=1;i<=block;i++) {
L[i]=R[i-1]+1; R[i]=block*i;
for(j=L[i];j<=R[i];j++) {
pos[j]=i;
b[i].fix(a[j].v,1);
}
}
if(R[block]!=n) {
block++; L[block]=R[block-1]+1; R[block]=n;
for(i=L[block];i<=R[block];i++) {
pos[i]=block;
b[block].fix(a[i].v,1);
}
}
for(i=1;i<=n;i++) {
ans+=(i-1-t.inq(a[i].v));
t.fix(a[i].v,1);
}
printf("%d\n",ans);
scanf("%d",&m);
while(m--) {
scanf("%d%d",&x,&y);
if(x>y) swap(x,y);
if(x==y||a[x].v==a[y].v) {
printf("%d\n",ans); continue;
}
if(a[x].v>a[y].v) ans--;
else ans++;
ans+=(query(x+1,y-1,a[y].v-1)+query(x+1,y-1,a[y].v)-query(x+1,y-1,a[x].v-1)-query(x+1,y-1,a[x].v));
int p=pos[x],q=pos[y];
if(p!=q) {
b[p].fix(a[x].v,-1); b[p].fix(a[y].v,1);
b[q].fix(a[y].v,-1); b[q].fix(a[x].v,1);
}
swap(a[x].v,a[y].v);
printf("%d\n",ans);
}
}