BZOJ 2038 小z的袜子

题意：（转） 作为一个生活散漫的人，小 Z 每天早上都要耗费很久从一堆五颜六色的袜子中找出一双来穿。终于有一天，小 Z 再也无法忍受这恼人的找袜子过程，于是他决定听天由命 ……
具体来说，小 Z 把这 N 只袜子从 1 到 N 编号，然后从编号 L 到 R(L  尽管小 Z 并不在意两只袜子是不是完整的一双，甚至不在意两只袜子是否一左一右，他却很在意袜子的颜色，毕竟穿两只不同色的袜子会很尴尬。

你的任务便是告诉小Z，他有多大的概率抽到两只颜色相同的袜子。当然，小Z希望这个概率尽量高，所以他可能会询问多个(L,R)以方便自己选择。

题解：莫队算法 分块乱搞

参考资料 ： http://www.cnblogs.com/kuangbin/p/3263483.html

代码：

/**************************************************************
    Problem: 2038
    User: alpc_wt
    Language: C++
    Result: Accepted
    Time:1176 ms
    Memory:3244 kb
****************************************************************/

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 50000 + 10;
int n,m,block;
ll ans = 0;
ll gcd(ll a,ll b)
{
    return a%b==0 ? b : gcd(b,a%b);
}
struct answer{
    int l,r,id;
    ll a,b;
}p[N];
int pos[N],col[N],s[N];
int cmp(answer a,answer b)
{
    if(pos[a.l] == pos[b.l]) return a.r < b.r;
    return a.l < b.l;
}
int cmp2(answer a,answer b)
{
    return a.id < b.id;
}
void update(int q,int x){
    ans -= s[col[q]]*s[col[q]];
    s[col[q]] += x;
    ans += s[col[q]]*s[col[q]];
}
void solve()
{
    int l,r;
    for(int i=1,l=1,r=0;i<=m;i++){//按块进行更新
        for(;r<p[i].r;r++)
            update(r+1,1);
        for(;r>p[i].r;r--)
            update(r,-1);
        for(;l<p[i].l;l++)
            update(l,-1);
        for(;l>p[i].l;l--)
            update(l-1,1);
        p[i].a = ans - (p[i].r - p[i].l + 1);
        p[i].b = (ll)(p[i].r - p[i].l + 1)*(p[i].r - p[i].l);
        if(p[i].a==0 || p[i].l==p[i].r){
            p[i].a = 0;
            p[i].b = 1;
            continue;
        }
        ll gg = gcd(p[i].a,p[i].b);
        p[i].a /= gg;
        p[i].b /= gg;
    }
}
int main()
{
    memset(s,0,sizeof(s));
    scanf("%d%d",&n,&m);
    block = (int)sqrt(n);
    for(int i=1;i<=n;i++) pos[i] = i/block + 1;
    for(int i=1;i<=n;i++) scanf("%d",&col[i]);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&p[i].l,&p[i].r);
        p[i].id = i;
    }
    sort(p+1,p+1+m,cmp);
    solve();
    sort(p+1,p+1+m,cmp2);
    for(int i=1;i<=m;i++)
        printf("%lld/%lld\n",p[i].a,p[i].b);
    return 0;
}