[BZOJ2038] [2009国家集训队]小Z的袜子(hose) 莫队算法练习

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

Time Limit: 20 Sec Memory Limit: 259 MB
Submit: 10299 Solved: 4685
[Submit][Status][Discuss]

Description

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

Input

输入文件第一行包含两个正整数N和M。N为袜子的数量，M为小Z所提的询问的数量。接下来一行包含N个正整数Ci，其中Ci表示第i只袜子的颜色，相同的颜色用相同的数字表示。再接下来M行，每行两个正整数L，R表示一个询问。

Output

包含M行，对于每个询问在一行中输出分数A/B表示从该询问的区间[L,R]中随机抽出两只袜子颜色相同的概率。若该概率为0则输出0/1，否则输出的A/B必须为最简分数。（详见样例）

Sample Input

6 4
1 2 3 3 3 2
2 6
1 3
3 5
1 6

Sample Output

2/5
0/1
1/1
4/15
【样例解释】
询问1：共C(5,2)=10种可能，其中抽出两个2有1种可能，抽出两个3有3种可能，概率为(1+3)/10=4/10=2/5。
询问2：共C(3,2)=3种可能，无法抽到颜色相同的袜子，概率为0/3=0/1。
询问3：共C(3,2)=3种可能，均为抽出两个3，概率为3/3=1/1。
注：上述C(a, b)表示组合数，组合数C(a, b)等价于在a个不同的物品中选取b个的选取方案数。
【数据规模和约定】
30%的数据中 N,M ≤ 5000；
60%的数据中 N,M ≤ 25000；
100%的数据中 N,M ≤ 50000，1 ≤ L < R ≤ N，Ci ≤ N。

表示蒟蒻并没有看出来是莫队（不熟练）。

先简述一下莫队算法，

莫队算法用于离线处理区间问题。当然它只能处理能在O（1）的时间从【l，r】转移到【l-1，r】【l+1，r】【l，r-1】【l，r+1】的问题。

具体思路如下：

设区间长度为n，

首先对查询进行分块，按照左端点的大小分成sqrt(n)块并按所属块排序，在每块内再按右端点的大小排序，之后从区间（0,0）一步一步移到排序好的下一个询问。然后将询问回复原序输出答案。

那么问题来了，时间复杂度是多少？

下面给出时间复杂度的证明：

首先，我们发现对于一个询问【l，r】我们能将其抽象为平面上的一个点（l,r）。

所以从一个询问走到下一个询问的时间是曼哈顿距离。

现在，我们已经分好了sqrt（n）块。

先考虑r，对于两个点，如果在同一个块，对于这两个点所处的块r单调，所以每一块r最多为O（n），由于有sqrt（n）个块，所以复杂度为O(n √n)

如果两个点在不同的块，r最多变化n，由于有√n块，所以复杂度为O(n√n)

再考虑l，如果两点在同一块，l变化不超过√n，如果两点不在同一块l变化同样不超过√n，由于有m个询问，所以时间复杂度为O（n√n）（n与m同级）

所以总复杂度为O（n√n）

下面是此题代码：

 #include<iostream>

 #include<cstdio>

 #include<cstdlib>

 #include<cstring>

 #include<cmath>

 #include<algorithm>

 #define LL long long

 using namespace std;

 LL n,m;

 LL c[];

 struct ask

 {

     LL l,r,id,a,b;

 }a[];

 LL belong[];

 LL size=;

 bool cmp(ask t1,ask t2)

 {

     if(belong[t1.l]==belong[t2.l]) return t1.r<t2.r;

     return t1.l<t2.l;

 }

 bool cmp1(ask t1,ask t2){return t1.id<t2.id;}

 LL ans=;

 LL s[];

 void update(int now,int add)

 {

     if(s[c[now]]>)

         ans-=s[c[now]]*(s[c[now]]-);

     s[c[now]]+=add;

     if(s[c[now]]>)

         ans+=s[c[now]]*(s[c[now]]-);

 }

 LL gcd(LL x,LL y){return y==?x:gcd(y,x%y);}

 void solve()

 {

     LL l=,r=;

     for(int i=;i<=m;i++)

     {

         for(;r<a[i].r;r++) update(r+,);

         for(;r>a[i].r;r--) update(r,-);

         for(;l<a[i].l;l++) update(l,-);

         for(;l>a[i].l;l--) update(l-,);

         if(r==l){a[i].a=,a[i].b=;continue;}

         a[i].a=ans,a[i].b=(r-l+)*(r-l);

         LL g=gcd(a[i].a,a[i].b);

         a[i].a/=g;

         a[i].b/=g;

     }

 }

 int main()

 {

     scanf("%lld%lld",&n,&m);

     size=sqrt(n);

     for(int i=;i<=n;i++) belong[i]=(i-)/size+;

     for(int i=;i<=n;i++) scanf("%lld",&c[i]);

     for(int i=;i<=m;i++)

     {

         scanf("%lld%lld",&a[i].l,&a[i].r);

         a[i].id=i;

     }

     sort(a+,a+m+,cmp);

     solve();

     sort(a+,a+m+,cmp1);

     for(int i=;i<=m;i++) printf("%lld/%lld\n",a[i].a,a[i].b);

 }

秒客网