【序列莫队】BZOJ2038- [2009国家集训队]小Z的袜子(hose)

【题目大意】
给出1-N只袜子的颜色，多次询问L-R中选出一双同色袜子的概率。

【思路】

裸莫队。基本的莫队步骤：
①分组（每组大小为根号sqrt(n)，共sqrt(n)组）
②排序（左边界分组，右边界在组内按大小排序）
③暴力转移

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=50000+50;
typedef long long ll;
struct node
{
    int l,r,id,pos;
    ll ansa,ansb;
};
int n,m,c[MAXN];
ll ans=0,s[MAXN];
ll sqr(ll x){return x*x;}  
node q[MAXN];
 
ll gcd(ll a,ll b)
{
    if (b==0) return a;
        else return gcd(b,a%b);
}
 
bool cmp(node a,node b)
{
    if (a.pos==b.pos) return (a.r<b.r);
        else return (a.pos<b.pos);
}
 
bool cmp_id(node a,node b)
{
    return (a.id<b.id);
}
 
void update(int x,int delta)
{
    ans-=sqr(s[c[x]]);
    s[c[x]]+=delta;
    ans+=sqr(s[c[x]]);
}
 
void init()
{
    memset(s,0,sizeof(s));
    memset(c,0,sizeof(c));
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%d",&c[i]);
    int block=int(sqrt(n));
    for (int i=0;i<m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].id=i;
        q[i].pos=(q[i].l-1)/block+1;
    }
    sort(q,q+m,cmp);
}
 
void solve()
{
    int l=1,r=1;
    s[c[1]]=1,ans=sqr(s[c[1]]);
    for (int t=0;t<m;t++)
    {
        for (;r<q[t].r;r++) update(r+1,1);
        for (;r>q[t].r;r--) update(r,-1);
        for (;l<q[t].l;l++) update(l,-1);
        for (;l>q[t].l;l--) update(l-1,1);
        if (l==r)
        {
            q[t].l=0;q[t].r=1;
            continue;
        }
        q[t].ansa=ans-(r-l+1);
        q[t].ansb=(ll)(r-l+1)*(r-l);
        ll minx=min(q[t].ansa,q[t].ansb),maxx=max(q[t].ansa,q[t].ansb);
        ll k=gcd(maxx,minx);
        q[t].ansa/=k,q[t].ansb/=k;
    }
}
 
void print()
{
    sort(q,q+m,cmp_id);
    for (int i=0;i<m;i++) printf("%lld/%lld
",q[i].ansa,q[i].ansb);
}
 
int main() 
{
    init();
    solve();
    print(); 
    return 0;
}