2038: [2009国家集训队]小Z的袜子(hose)

题目链接

题目大意：求在区间[L,R]中选2个数字，数字相同的概率

题解：莫队算法
处理问题：不带修改的区间询问(加上时间倒流可以处理简单修改)
使用要求：[l-1,r] [l,r-1]的结果可由[l,r]的答案在O(1)或O(logn)的时间内推出
对于这道题来说，只需要知道C(n,2)=n*(n-1)/2就可以做啦

我的收获：复制了莫队模板(雾

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define M 50500

int now,d,z;
int n,m,blo;
int a[M],pos[M],cnt[M],anx[M],any[M];

struct node{int l,r,id;}q[M];

bool cmp(node x,node y){return pos[x.l]<pos[y.l]||(pos[x.l]==pos[y.l]&&x.r<y.r);}

void update(int x,int v)
{
    now-=cnt[x]*(cnt[x]-1)/2;//C(cnt[x],2)
    cnt[x]+=v;
    now+=cnt[x]*(cnt[x]-1)/2;
}

inline int calc(unsigned x,unsigned y){return (y-x+1)*(y-x)/2;}//防爆int 

void work()
{
    int l=1,r=0;
    for(int i=1;i<=m;i++)
    {
        while(r<q[i].r) update(a[++r],1);
        while(l>q[i].l) update(a[--l],1);
        while(r>q[i].r) update(a[r--],-1);
        while(l<q[i].l) update(a[l++],-1);
        if(now==0) {anx[q[i].id]=0,any[q[i].id]=1;continue;}
        z=calc(q[i].l,q[i].r);d=__gcd(now,z);
        anx[q[i].id]=now/d;any[q[i].id]=z/d;
    }
    for(int i=1;i<=m;i++)
        printf("%d/%d\n",anx[i],any[i]);
}

void init()
{
    cin>>n>>m;blo=sqrt(n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        pos[i]=(i-1)/blo+1;
    }
    for(int i=1;i<=m;i++)
        scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
    sort(q+1,q+m+1,cmp);
}

int main()
{
    init();
    work();
    return 0;
}