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

Description

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

Solution

莫队算法的入门题咯
以前一直觉得莫队好玄乎，今天重新看才发现其实蛮简单。。
就是把询问重新排序分个块呗。。
看完论文信心满满去写这题。
看完题目啪啪啪敲了个线段树，然后飞速TLE==
然后开始钦佩自己==往标算上套个log只是分分钟的事情咯

Code

/**************************************************************    Problem: 2038    User: bblss123    Language: C++    Result: Accepted    Time:2088 ms    Memory:3256 kb****************************************************************/#include<iostream>#include<string.h>#include<algorithm>#include<stdio.h>#include<math.h>using namespace std;typedef long long ll;const int M=5e4+5;int cnt[M],n,m,c[M],S;struct Query{    int l,r,id,pl;    inline bool operator < (const Query &tmp)const{        return pl<tmp.pl||pl==tmp.pl&&r<tmp.r;    }}ques[M];inline void rd(int &a){    a=0;char c;    while(c=getchar(),!isdigit(c));    do a=a*10+(c^48);        while(c=getchar(),isdigit(c));}struct grad{ll x,y;}grade[M];inline ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}ll res=0;inline void modify(int pos,int ct){    res-=1ll*cnt[c[pos]]*(cnt[c[pos]]-1);    cnt[c[pos]]+=ct;    res+=1ll*cnt[c[pos]]*(cnt[c[pos]]-1);}inline void gao(){    int l=0,r=-1;    for(int i=0;i<m;++i){        for(;l<ques[i].l;++l)modify(l,-1);        for(;l>ques[i].l;--l)modify(l-1,1);        for(;r>ques[i].r;--r)modify(r,-1);        for(;r<ques[i].r;++r)modify(r+1,1);        if(ques[i].l==ques[i].r){            grade[ques[i].id]=(grad){0,1};            continue;        }        ll w=1ll*(ques[i].r-ques[i].l)*(ques[i].r-ques[i].l+1);        int gd=gcd(res,w);        grade[ques[i].id].x=res/gd,grade[ques[i].id].y=w/gd;    }}int main(){    cin>>n>>m;    for(int i=0;i<n;++i)        scanf("%d",&c[i]);    S=(int)sqrt(n+1)+1;    for(int i=0,l,r;i<m;++i){        rd(l),rd(r);        ques[i]=(Query){l-1,r-1,i,(l-1)/S};    }    sort(ques,ques+m);    gao();    for(int i=0;i<m;++i)        printf("%lld/%lld\n",grade[i].x,grade[i].y);    return 0;}