【2010集训队出题】小Z的袜子

Description

　作为一个生活散漫的人，小Z每天早上都要耗费很久从一堆五颜六色的袜子中找出一双来穿。终于有一天，小Z再也无法忍受这恼人的找袜子过程，于是他决定听天由命……

　　具体来说，小Z把这N只袜子从1到N编号，然后从编号L到R(L 尽管小Z并不在意两只袜子是不是完整的一双，甚至不在意两只袜子是否一左一右，他却很在意袜子的颜色，毕竟穿两只不同色的袜子会很尴尬。

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

Solution

一道莫队算法的裸题。莫队算法学习小记

设一个T[i]表示上一个询问的i出现的次数，每次遇到一个a就T[a]++;

ans=∑ni=1C2T[i]C2r?l+1=∑ni=1T[i]2?∑ni=nT[i](r?l+1)(r?l)=∑ni=1T[i]2?(r?l+1)(r?l+1)(r?l)

每次暴力维护ans就好了。

Solution

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=50007;
typedef long long ll;
ll i,j,k,l,t,n,m,r,ans1,kuai;
ll ans[maxn][2],T[maxn];
ll c[maxn];
struct node{
    ll a,b,c,d,e;
}a[maxn];
bool cmp(node x,node y){
    return x.d<y.d||x.d==y.d&&x.e<y.e;
}
void update(ll l,ll r,ll z){
    int j;
    fo(j,l,r)ans1-=T[c[j]]*T[c[j]],T[c[j]]+=z,ans1+=T[c[j]]*T[c[j]];
}
ll gcd(ll x,ll y){return (y)?gcd(y,x%y):x;}
int main(){
    scanf("%lld%lld",&n,&m);
    kuai=sqrt(n);
    fo(i,1,n)scanf("%lld",&c[i]);
    fo(i,1,m){
        scanf("%lld%lld",&a[i].a,&a[i].b);
        a[i].d=(a[i].a-1)/kuai+1;
        a[i].e=(a[i].b-1)/kuai+1;
        a[i].c=i;
    }
    sort(a+1,a+1+m,cmp);
    l=1;r=0;
    fo(i,1,m){
        k=a[i].a;t=a[i].b;
        if(k>l)update(l,k-1,-1);
        else if(k<l)update(k,l-1,1);
        if(t>r)update(r+1,t,1);
        else if(r>t)update(t+1,r,-1);
        l=k;r=t;
        ans[a[i].c][0]=ans1-(t-k+1);
        ans[a[i].c][1]=(t-k+1)*(t-k);
        if(ans[a[i].c][0]==0){
            ans[a[i].c][0]=0;ans[a[i].c][1]=1;
        }
        else{
            ll u=gcd(ans[a[i].c][0],ans[a[i].c][1]);
            ans[a[i].c][0]/=u;
            ans[a[i].c][1]/=u;
        }
    }
    fo(i,1,m)printf("%lld/%lld\n",ans[i][0],ans[i][1]);
}