[BZOJ2038]小Z的袜子

来源：互联网发布：阿里算法平台编辑：程序博客网时间：2024/05/19 22:46

原题地址

莫队出的莫队算法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;}

0 0