莫队算法-小Z的袜子

小Z的妹子袜子这道题用的是莫队算法，据说解决离线区间询问几乎无敌。

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

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

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

对于一个区间l，r，如果其中有v种颜色，$f[v_i]$表示第i种颜色，有多少个袜子。那么l，r的概率就是：

$\LARGE\frac{\sum_{i=1}^{v}C_2^{f_i}}{C_2^{r-l+1}}=\frac{\sum_{i=1}^vf_i^2-\sum_{i=1}^{v}f_i}{C_2^{r-l+1}}=\frac{\sum_{i=1}^{v}f_i^2-(r-l+1)}{2C_2^{r-l+1}}$

，我们发现，除了$\sum_{i=1}^{v}f_i^2$，其它都可以方便求出。那么怎么求这个各个颜色的袜子数目的平方和呢？

　　对于区间l，r，如果询问是l，r+1，我们发现$S_现=S_原-f_x^2+(f_x+1)^2=S_原+2f_x+1$。我们只需加上$2f_x+1$即可。这启发我们遇到一个新询问区间，都可以通过一个个加过来解决。只需维护l，r两个指针即可。

　　但是，对于随机数据，这样的做法显然是$n^2$的。于是就有一个神奇的莫队算法：把所有查询都根据左端点的pos分块，然后排序。如果左端点位于同一个块，就按照右端点排序，然后l，r再移动，复杂度就变成了$n\sqrt{n}$。。

　　为什么呢？因为左端点在一个块时，右端点最多移动$\sqrt{n}$次，而左端点最多跳$\sqrt{n}*\sqrt{n}$次，而有$\sqrt{n}$个这样的块，这样左端点时间复杂度比右端点大，所以在一个块中的时间复杂度是n，乘上有$\sqrt{n}$个块，就是$O(n\sqrt{n})$。而一共有$\sqrt{n}$次左端点跨越块，右端点每次这样移动n，左端点每次移动$\sqrt{n}$，也就是说右端点时间复杂度更大，同样时间复杂度也是$O(n\sqrt{n})$。

 1 #include <cmath>
 2 #include <cstdio>
 3 #include <algorithm>
 4 using namespace std;
 5 
 6 typedef long long LL;
 7 const int maxn=5e4+5;
 8 int n, m, l=1, r, ans;
 9 int color[maxn], f[maxn];
10 
11 LL gcd(LL x, LL y){ return y?gcd(y, x%y):x; }
12 
13 struct query{
14     int l, r, pos;
15     //又双叒叕忘记LL了
16     int id; LL ans;
17 }q[maxn];
18 
19 bool cmp_md(query a, query b){
20     return a.pos!=b.pos?a.pos<b.pos:a.r<b.r; }
21 bool cmp_id(query a, query b){
22     return a.id<b.id; }
23 
24 void modify(int now, int shift){
25     //别忘记+1!
26     ans+=2*shift*f[color[now]]+1;
27     f[color[now]]+=shift;
28 }
29 
30 int main(){
31     scanf("%d%d", &n, &m);
32     for (int i=1; i<=n; ++i) scanf("%d", &color[i]);
33     int sqrtn=sqrt(double(n));
34     for (int i=0; i<m; ++i){
35         scanf("%d%d", &q[i].l, &q[i].r);
36         q[i].pos=q[i].l/sqrtn; q[i].id=i;
37     }
38     sort(q, q+m, cmp_md);
39     for (int i=0; i<m; ++i){
40         //这里要是反过来的!
41         //注意拓展和删除的区别..
42         if (q[i].l<l) for (; l!=q[i].l; --l) modify(l-1, 1);
43         if (q[i].l>l) for (; l!=q[i].l; ++l) modify(l, -1);
44         if (q[i].r>r) for (; r!=q[i].r; ++r) modify(r+1, 1);
45         if (q[i].r<r) for (; r!=q[i].r; --r) modify(r, -1);
46         q[i].ans=ans;
47     }
48     sort(q, q+m, cmp_id);
49     LL x, y;
50     for (int i=0; i<m; ++i){
51         x=q[i].ans-(q[i].r-q[i].l+1);
52         LL t=q[i].r-q[i].l+1;
53         y=t*(t-1);
54         int g=gcd(x, y);
55         x/=g, y/=g;
56         printf("%lld/%lld\n", x, y);
57     }
58     return 0;
59 }

 

转载于:https://www.cnblogs.com/MyNameIsPc/p/7563012.html