[BZOJ2038]小Z的袜子

原题地址

莫队出的莫队算法OLZ

AC code:

#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
const int N=100010;
int n,m,k,tot,g;
int a[N],c[N],app[N],ans1[N],ans2[N];

struct inq
{
    int l,r,num;
}q[N];

bool cmp(inq x,inq y)
{
    if(x.l/m!=y.l/m) return x.l/m<y.l/m;
    return x.r<y.r;
}

int gcd(int x,int y)
{
    return y?gcd(y,x%y):x;
}

int main()
{
    scanf("%d%d",&n,&k);
    m=(int)sqrt(n);
    for(int i=2;i<=n;i++) c[i]=((long long)i*(i-1))>>1;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=k;i++){
        q[i].num=i;
        scanf("%d%d",&q[i].l,&q[i].r);
    }
    sort(q+1,q+k+1,cmp);
    for(int i=q[1].l;i<=q[1].r;i++) tot+=app[a[i]]++;
    g=gcd(tot,c[q[1].r-q[1].l+1]);
    ans1[q[1].num]=tot/g;
    ans2[q[1].num]=c[q[1].r-q[1].l+1]/g;
    for(int i=2;i<=k;i++){
        if(q[i].l<q[i-1].l){
            for(int j=q[i].l;j<q[i-1].l;j++) tot+=app[a[j]]++;
        }
        else{
            for(int j=q[i-1].l;j<q[i].l;j++) tot-=--app[a[j]];
        }
        if(q[i].r>q[i-1].r){
            for(int j=q[i-1].r+1;j<=q[i].r;j++) tot+=app[a[j]]++;
        }
        else{
            for(int j=q[i].r+1;j<=q[i-1].r;j++) tot-=--app[a[j]];
        }
        g=gcd(tot,c[q[i].r-q[i].l+1]);
        ans1[q[i].num]=tot/g;
        ans2[q[i].num]=c[q[i].r-q[i].l+1]/g;
    }
    for(int i=1;i<=k;i++) printf("%d/%d\n",ans1[i],ans2[i]);

    return 0;
}