[BZOJ2141]排队（分块+树状数组求逆序对）

题目描述

传送门

题解

每次修改维护序列逆序对的个数。
用分块来搞。
考虑一下进行一次修改对于答案的影响。对于l来说，交换之后应该减去l～r中比l小的数的个数，加上l～r中比l大的数的个数；对于r来说，交换之后应该减去l～r中比r大的个数，加上l～r中比r小的个数。
这样的话，块外暴力找，块内sort之后二分，修改暴力重构。
我直接把lr包括在范围内做了，所以会有一些吃屎的判断来去重。如果不包括lr只做它们之间的数可能会更简单一点。

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;

const int max_n=2e4+5;
const int max_t=2e2+5;

int n,m,cnt,t1,t2,l,r,ans;
int a[max_n],p[max_n];
int h[max_n],hh[max_n],num[max_n],last[max_t];
int C[max_n];

inline int in(){
    int x=0; char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}

inline int cmp(int x,int y){
    return a[x]<a[y];
}

inline void add(int loc,int val){
    for (int i=loc;i<=n;i+=i&(-i))
      C[i]+=val;
}

inline int query(int loc){
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
      ans+=C[i];
    return ans;
}

inline void BIT(){
    for (int i=n;i>=1;--i){
        ans+=query(h[i]-1);
        add(h[i],1);
    }
    printf("%d\n",ans);
}

inline void solve(int l,int r){
    int lrange,rrange;
    int ans1=0,ans2=0,ans3=0,ans4=0;
    if (num[l]==num[r]){
        for (int i=l;i<=r;++i){
            if (h[i]>h[l]) ++ans1;
            if (h[i]>h[r]) ++ans2;
            if (h[i]<h[l]) ++ans3;
            if (h[i]<h[r]) ++ans4;
        }
        if (h[l]<h[r]) {if (ans4) --ans4;}
        else {if (ans3) --ans3;}
        ans+=ans1-ans2-ans3+ans4;
        printf("%d\n",ans);

        swap(h[l],h[r]);
        return;
    }

    if (l!=last[num[l]-1]+1){
        lrange=num[l]+1;
        for (int i=l;i<=last[num[l]];++i){
            if (h[i]>h[l]) ++ans1;
            if (h[i]>h[r]) ++ans2;
            if (h[i]<h[l]) ++ans3;
            if (h[i]<h[r]) ++ans4;
        }
    }
    else lrange=num[l];

    if (r!=last[num[r]]){
        rrange=num[r]-1;
        for (int i=last[num[r]-1]+1;i<=r;++i){
            if (h[i]>h[l]) ++ans1;
            if (h[i]>h[r]) ++ans2;
            if (h[i]<h[l]) ++ans3;
            if (h[i]<h[r]) ++ans4;
        }
    }
    else rrange=num[r];

    for (int i=lrange;i<=rrange;++i){
        int loc1=upper_bound(hh+last[i-1]+1,hh+last[i]+1,h[l])-hh;
        int loc2=upper_bound(hh+last[i-1]+1,hh+last[i]+1,h[r])-hh;
        int loc3=lower_bound(hh+last[i-1]+1,hh+last[i]+1,h[l])-hh-1;
        int loc4=lower_bound(hh+last[i-1]+1,hh+last[i]+1,h[r])-hh-1;
        if (loc1<=last[i]) ans1+=last[i]-loc1+1;
        if (loc2<=last[i]) ans2+=last[i]-loc2+1;
        if (loc3>=last[i-1]+1) ans3+=loc3-(last[i-1]+1)+1;
        if (loc4>=last[i-1]+1) ans4+=loc4-(last[i-1]+1)+1;
    }
    if (h[l]<h[r]) {if (ans4) --ans4;}
    else {if (ans3) --ans3;}
    ans+=ans1-ans2-ans3+ans4;
    printf("%d\n",ans);

    swap(h[l],h[r]);
    for (int i=last[num[l]-1]+1;i<=last[num[l]];++i)
      hh[i]=h[i];
    sort(hh+last[num[l]-1]+1,hh+last[num[l]]+1);
    for (int i=last[num[r]-1]+1;i<=last[num[r]];++i)
      hh[i]=h[i];
    sort(hh+last[num[r]-1]+1,hh+last[num[r]]+1);
}

int main(){
    n=in();
    for (int i=1;i<=n;++i){
        a[i]=in();
        p[i]=i;
    }
    sort(p+1,p+n+1,cmp);
    for (int i=1;i<=n;++i){
        if (a[p[i]]!=a[p[i-1]]) h[p[i]]=++cnt;
        else h[p[i]]=cnt;
    }
    for (int i=1;i<=n;++i)
      hh[i]=h[i];

    ans=0;
    BIT();  

    t1=(int)sqrt(n);
    if (n%t1) t2=n/t1+1;
    else t2=n/t1;
    for (int i=1;i<=n;++i)
      num[i]=(i-1)/t1+1;
    for (int i=1;i<=t2;++i){
        last[i]=i*t1;
        if (last[i]>n) last[i]=n;
    }
    for (int i=1;i<=t2;++i)
      sort(hh+last[i-1]+1,hh+last[i]+1);

    m=in();
    for (int i=1;i<=m;++i){
        l=in(); r=in();
        if (l==r||h[l]==h[r]){
            printf("%d\n",ans);
            continue;
        }
        if (l>r) swap(l,r);
        solve(l,r);
    }
}