bzoj 2141 排队 【分块】

http://www.lydsy.com/JudgeOnline/problem.php?id=2141
排排坐，吃果果，生果甜嗦嗦，大家笑呵呵。你一个，我一个，大的分给你，小的留给我，吃完果果唱支歌，大家
乐和和。红星幼儿园的小朋友们排起了长长地队伍，准备吃果果。不过因为小朋友们的身高有所区别，排成的队伍
高低错乱，极不美观。设第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

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+5;
int a[maxn],b[maxn],c[maxn],d[maxn],be[maxn];
int n,m,block_len,block_cnt;
long long ans;
inline void init()
{
    block_len=sqrt(n);
    if(n%block_len) block_cnt=n/block_len+1;
    else block_cnt=n/block_len;
    for(int i=1;i<=n;i++){
        be[i]=(i-1)/block_len+1;
    }
}
inline void add(int pos,int val)
{
    while(pos<=n){
        c[pos]+=val;
        pos+=pos&(-pos);
    }
}
inline int getsum(int pos)
{
    int sum=0;
    while(pos>0){
        sum+=c[pos];
        pos-=pos&(-pos);
    }
    return sum;
}
inline void rebuild(int p)
{
    int l=(p-1)*block_len+1;
    int r=min(n,p*block_len);
    for(int i=l;i<=r;i++){
        b[i]=a[i];
    }
    sort(b+l,b+r+1);
}
inline void pre()
{
    for(int i=n;i>=1;i--){
        ans+=getsum(a[i]-1);
        add(a[i],1);
    }
    for(int i=1;i<=block_cnt;i++){
        rebuild(i);
    }
    printf("%lld\n",ans);
}
void solve(int l,int r)
{
    if(l>r) swap(l,r);
    if(a[l]>a[r]) ans--;
    if(a[l]<a[r]) ans++;
    if(a[l]!=a[r]){
        if(be[l]==be[r]){
            for(int i=l+1;i<r;i++){
                if(a[i]>a[l]) ans++;
                if(a[i]<a[l]) ans--;
                if(a[i]>a[r]) ans--;
                if(a[i]<a[r]) ans++;
            }
        }
        else{
            for(int i=l+1;i<=be[l]*block_len;i++){
                if(a[i]>a[l]) ans++;
                if(a[i]<a[l]) ans--;
                if(a[i]>a[r]) ans--;
                if(a[i]<a[r]) ans++;
            }
            for(int i=(be[r]-1)*block_len+1;i<r;i++){
                if(a[i]>a[l]) ans++;
                if(a[i]<a[l]) ans--;
                if(a[i]>a[r]) ans--;
                if(a[i]<a[r]) ans++;
            }
            for(int i=be[l]+1;i<=be[r]-1;i++){
                int L=(i-1)*block_len+1;
                int R=min(n,i*block_len);
                int temp;
                temp=upper_bound(b+L,b+R+1,a[l])-b-L;
                ans+=block_len-temp;
                temp=lower_bound(b+L,b+R+1,a[l])-b-L;
                ans-=temp;

                temp=upper_bound(b+L,b+R+1,a[r])-b-L;
                ans-=block_len-temp;
                temp=lower_bound(b+L,b+R+1,a[r])-b-L;
                ans+=temp;
            }
        }
    }
    swap(a[l],a[r]);
    rebuild(be[l]);
    rebuild(be[r]);
    printf("%lld\n",ans);
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        d[i]=a[i];
    }
    sort(d+1,d+n+1);
    for(int i=1;i<=n;i++){
        a[i]=b[i]=lower_bound(d+1,d+n+1,a[i])-d;
    }
    init();
    pre();
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        int l,r;
        scanf("%d%d",&l,&r);
        solve(l,r);
        //for(int i=1;i<=n;i++) printf("%d ",d[a[i]]); puts("");
    }
    return 0;
}