关闭

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

标签: ACM
213人阅读 评论(0) 收藏 举报
分类:

莫队算法的经典题,在一个区间内,维护一定颜色的袜子的数量即可,复杂度: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]);
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:83668次
    • 积分:3063
    • 等级:
    • 排名:第11375名
    • 原创:230篇
    • 转载:0篇
    • 译文:0篇
    • 评论:14条