bzoj 2038 [2009国家集训队]小Z的袜子 莫队算法

莫队算法的经典题，在一个区间内，维护一定颜色的袜子的数量即可，复杂度：O（nsqrt（n））；

#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>#include<iostream>using namespace std;const int maxn=50005;int bit[maxn];int a[maxn],b[maxn];typedef long long LL;LL s1[maxn];int low(int p){    return  p&(-p);}void merg(int p,int n,int k){    while(p<=n){        bit[p]+=k;        p=p+low(p);    }}int sum(int p){    int s=0;    while(p>0){        s+=bit[p];        p=p-low(p);    }    return s;}struct pi{    int l,r,x;    int id;}pp[maxn];int cmp(pi a,pi b){    if(a.x!=b.x) return a.x<b.x;    return a.r<b.r;}int main(){    int i,j,n,m;    cin>>n;    for(i=1;i<=n;i++){        scanf("%d",&a[i]);        b[i]=a[i];    }    sort(b+1,b+1+n);    for(i=1;i<=n;i++) a[i]=lower_bound(b+1,b+1+n,a[i])-b;    cin>>m;    int p=sqrt(n);    for(i=0;i<m;i++){        scanf("%d%d",&pp[i].l,&pp[i].r);        pp[i].id=i;        pp[i].x=(pp[i].l-1)/p+1;    }    sort(pp,pp+m,cmp);    LL s=0;    int l=0,r=0;    for(i=0;i<m;i++){        if(i==0){            for(j=pp[i].l;j<=pp[i].r;j++){                s+=sum(n)-sum(a[j]);                merg(a[j],n,1);            }        }        else{        if(l<pp[i].l){            for(j=l;j<pp[i].l;j++){                p=sum(a[j]-1);                if(p>0)                s-=p;                merg(a[j],n,-1);            }        }        else{            for(j=l-1;j>=pp[i].l;j--){                p=sum(a[j]-1);                if(p>0)                s+=p;                merg(a[j],n,1);            }        }        if(r<pp[i].r){            for(j=r+1;j<=pp[i].r;j++){                p=sum(n)-sum(a[j]);                if(p>0) s+=p;                merg(a[j],n,1);            }        }        else{            for(j=r;j>pp[i].r;j--){                p=sum(n)-sum(a[j]);                if(p>0) s-=p;                merg(a[j],n,-1);            }        }        }        s1[pp[i].id]=s;        l=pp[i].l;        r=pp[i].r;    }    for(i=0;i<m;i++) printf("%lld\n",s1[i]);}