CodeForces - 785E Anton and Permutation

题面

题意

有一个1-n的排列，初始为1,2,3……n，每次交换其中两个数，并且询问此时逆序对的个数。

做法

某次重新计算逆序对的个数显然会T，因此肯定是要每次计算交换之后对答案产生的影响。
经过观察，交换p，q两位置的数，对答案的影响仅对数字大小在num[p],num[q]之间的数的个数有关。
因此可以考虑分块，对于交换同一块里的两个数时，暴力修改即可。
若在不同的块内，则先暴力计算p，q所在的两个块对答案的影响，然后对于中间的每个块，新维护一个数组，这个数组在每个块内的数与原来的数一样，但顺序不同，单调递增，查询这个块时，只要二分这个序列即可。
维护这个序列也十分简单，每次交换时，进行两次插入排序即可。

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#define ll long long
#define N 200100
#define M 510
#define Q 50010
using namespace std;

ll n,m,s,num[N],a[N],be[N],le[M],ri[M],id[N],ans,tmp;

inline ll ask(ll u,ll v,ll w)
{
    ll l,r,mid;
    for(l=u,r=v+1;l<r;)
    {
        mid=((l+r)>>1);
        if(a[mid]<w) l=mid+1;
        else r=mid;
    }
    return l-u;
}

int main()
{
    ll i,j,p,q,t1,t2;
    cin>>n>>m;
    s=sqrt(n);
    for(i=1;i<=n;i++)
    {
        a[i]=num[i]=id[i]=i;
        be[i]=i/s+1;
        if(be[i]>be[i-1])
        {
            le[be[i]]=i;
            ri[be[i-1]]=i-1;
        }
    }
    while(m--)
    {
        scanf("%lld%lld",&p,&q);
        if(p==q)
        {
            printf("%lld\n",ans);
            continue;
        }
        if(p>q) swap(p,q);
        num[p]<num[q]?ans++:ans--;
        if(be[p]==be[q])
        {
            for(i=p+1;i<q;i++)
            {
                if(num[p]>num[i]&&num[i]>num[q]) ans-=2;
                if(num[p]<num[i]&&num[i]<num[q]) ans+=2;
            }
            swap(num[p],num[q]);
            printf("%lld\n",ans);
            continue;
        }
        else
        {
            for(i=p+1;be[i]==be[i-1];i++)
            {
                if(num[p]>num[i]&&num[i]>num[q]) ans-=2;
                if(num[p]<num[i]&&num[i]<num[q]) ans+=2;
            }
            for(i=q-1;be[i]==be[i+1];i--)
            {
                if(num[p]>num[i]&&num[i]>num[q]) ans-=2;
                if(num[p]<num[i]&&num[i]<num[q]) ans+=2;
            }
            for(i=be[p]+1;i<be[q];i++)
            {
                ans+=(ask(le[i],ri[i],num[q])-ask(le[i],ri[i],num[p]))*2;
            }
            printf("%lld\n",ans);
            t1=id[num[p]],t2=id[num[q]];
            swap(num[p],num[q]);
            swap(id[num[p]],id[num[q]]);
            p=t1,q=t2;
            swap(a[p],a[q]);
            for(;a[p]<a[p-1]&&be[p]==be[p-1];p--) swap(a[p],a[p-1]),swap(id[a[p]],id[a[p-1]]);
            for(;a[p]>a[p+1]&&be[p]==be[p+1];p++) swap(a[p],a[p+1]),swap(id[a[p]],id[a[p+1]]);
            for(;a[q]<a[q-1]&&be[q]==be[q-1];q--) swap(a[q],a[q-1]),swap(id[a[q]],id[a[q-1]]);
            for(;a[q]>a[q+1]&&be[q]==be[q+1];q++) swap(a[q],a[q+1]),swap(id[a[q]],id[a[q+1]]);
        }
    }
}