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

Description
作为一个生活散漫的人，小Z每天早上都要耗费很久从一堆五颜六色的袜子中找出一双来穿。终于有一天，小Z再也无法忍受这恼人的找袜子过程，于是他决定听天由命……
具体来说，小Z把这N只袜子从1到N编号，然后从编号L到R(L 尽管小Z并不在意两只袜子是不是完整的一双，甚至不在意两只袜子是否一左一右，他却很在意袜子的颜色，毕竟穿两只不同色的袜子会很尴尬。
你的任务便是告诉小Z，他有多大的概率抽到两只颜色相同的袜子。当然，小Z希望这个概率尽量高，所以他可能会询问多个(L,R)以方便自己选择。

---

【题目分析】
分块+莫队

---

【代码】

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int n,m,w,tot;
int a[50001],l[50001],r[50001],rank[50001];
int num[50001],st[50001],en[50001];
long long ans=0;
long long up[50001];
long long down[50001];
inline long long gcd(long long a,long long b)
{return b==0?a:gcd(b,a%b);}
inline bool cmp1(int a,int b)
{return l[a]==l[b]?r[a]<r[b]:l[a]<l[b];}
inline bool cmp2(int a,int b)
{return r[a]==r[b]?l[a]<l[b]:r[a]<r[b];}
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
inline void mod(int pos,int f)
{
    ans-=(long long)num[a[pos]]*(num[a[pos]]-1)/2;
    num[a[pos]]+=f;
    ans+=(long long)num[a[pos]]*(num[a[pos]]-1)/2;
}
inline void print(long long a,long long b)
{
    if (a==0) {printf("0/1\n"); return;}
    long long g=gcd(a,b);
    printf("%lld/%lld\n",a/g,b/g);  
}
inline void solve(int be,int ed)
{
    int ll=l[rank[be]],rr=r[rank[be]];
    for (int i=ll;i<=rr;++i) mod(i,1);
    up[rank[be]]=ans; down[rank[be]]=(long long)(rr-ll+1)*(rr-ll)/2;
    for (int i=be+1;i<=ed;++i)
    {
        for (;ll>l[rank[i]];ll--) mod(ll-1,1);
        for (;ll<l[rank[i]];ll++) mod(ll,-1);
        for (;rr>r[rank[i]];rr--) mod(rr,-1);
        for (;rr<r[rank[i]];rr++) mod(rr+1,1);
        up[rank[i]]=ans;
        down[rank[i]]=(long long)(rr-ll+1)*(rr-ll)/2;
    }
}
int main()
{
    n=read();m=read();
    for (int i=1;i<=n;++i) a[i]=read();
    for (int i=1;i<=m;++i)
        l[i]=read(),r[i]=read(),rank[i]=i;
    sort(rank+1,rank+m+1,cmp1);
    w=sqrt(n);
    for (int i=1;i<=n;i+=w)
    {
        st[++tot]=i;
        en[tot]=i+w-1;
    }
    int ll=1,rr=0;
    for (int i=1;i<=tot;++i)
    {
        ans=0;
        memset(num,0,sizeof num);
        while (l[rank[ll]]<st[i]&&ll<=m) ll++;
        while (l[rank[rr]]<=en[i]&&rr<=m) rr++;
        if (ll<rr)
        {
            sort(rank+ll,rank+rr,cmp2);
            solve(ll,rr-1);
        }
    }
    for (int i=1;i<=m;++i)
        print(up[i],down[i]);
}