BZOJ2038

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=2038

题意：有N个袜子，每个袜子有不同的颜色，给出M个询问，每次询问给出L，R，从L~R区间选择2个袜子，问2个袜子相同的概率，输出格式见题目。

题解：轻松推下转化为求L,R里，设有K个不同的数，每个数出现的次数为a[i]。答案为 sigma(1~k)a[i]^2 - (r-l+1）/ (r-l+1)^2。

这里用莫队维护这个操作，复杂度T(N) = O（N^1.5）;

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N = 5E4 + 7;
int c[N], Be[N], n, q;
ll sum[N], ans; //ans为当前结果，sum[c[i]]为第c[i]种颜色到目前为止出现的次数。
struct node
{
    int l, r, id;
    ll a, b;
}Q[N];
bool cmp1(const node & x, const node & y) //先按L排序，L所在块数一样，再按R排序
{
    if(Be[x.l] == Be[y.l]) return x.r < y.r;
    return x.l < y.l;
}
bool cmp2(const node & x, const node & y) //为了输出答案。
{
    return x.id < y.id;
}
ll squ(ll x){return x * x;}
void init()
{
    int BLOCK = sqrt(n); //sqrt(n)为分块
    for(int i = 1; i <= n; i ++) scanf("%d",&c[i]), Be[i] = (i - 1) / BLOCK + 1;
    for(int i = 1; i <= q; i ++) scanf("%d%d",&Q[i].l, &Q[i].r), Q[i].id = i;
    sort(Q+1, Q+1+q, cmp1);
}
void modify(int p, int add)
{
    ans -= squ(sum[c[p]]);
    sum[c[p]] += add;
    ans += squ(sum[c[p]]);
}
void solve()
{
    for(int i = 1, l = 1, r = 0; i <= q; i ++) {
        while(Q[i].r > r) modify(r + 1, 1), r ++; //当前询问的r，比已经算的r大，继续向后求结果。
        while(Q[i].r < r) modify(r, -1), r --; //当前询问的r, 比已经算的r小， 减去前面的。
        while(Q[i].l > l) modify(l, -1), l ++; //
        while(Q[i].l < l) modify(l - 1, 1), l --;
        int len = Q[i].r - Q[i].l + 1;
        if(len == 1) {
            Q[i].a = 0, Q[i].b = 1;
        } else {
            Q[i].a = ans - len;
            Q[i].b = len * 1ll * (len - 1);
            ll d = __gcd(Q[i].a, Q[i].b);
            Q[i].a /= d, Q[i].b /= d;
        }
    }
    sort(Q+1, Q+1+q, cmp2);
    for(int i = 1; i <= q; i ++) printf("%lld/%lld\n",Q[i].a, Q[i].b);
}
int main()
{
    scanf("%d%d",&n,&q);
    init();
    solve();
    return 0;
}