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;
}