Codechef DECEMBER16 BOUNCE

题目大意

假设现在有个$R*C$的盒子，一开始在$(0,0)$有颗珠子，你朝右上角45度方向把珠子弹出，珠子碰到墙壁会反弹（依据反射定律），珠子到了某个角后就弹出了，游戏结束。珠子每到一个墙，就会相对应的把墙上的字符记录到$S(R,C)$上，比如珠子先后碰到上下左右，那么$S(R,C)="TDLR"$，一个盒子的权值就是$S(R,C)$的长度。
现在给定一个字符串$S$，以及一个正整数$N$，问所有满足$S$为$S(R,C)$的前缀，且$1 \leq R,C \leq N$的盒子的权值之和对$10^9+7$取模的值。

数据范围

$N \leq 10^{10},|S| \leq 10^6$

题解

首先我们要观察一些东西，记$f(R,C)$表示$R*C$的矩阵的$|S|$，那么可以观察到，若$(R,C)=1,f(R,C)=R+C-2$，否则，$f(R,C)=f(\frac{R}{(R,C)},\frac{C}{(R,C)})$。接下来，对于那个输入的S的限制，我们可以理解为有个点，他有两个势能，记为$(a,b)$，初始为$(R,C)$，那么假如当前弹到左右两面墙，相当于$a<b$，然后势能变成$(R,b-a)$，否则相当于$a>b$，势能变为$(a-b,C)$，那么我们可以搞出来一些约束，形如$aR>bC$以及$xR<yC$这样，找到约束最紧的两个。现在考虑怎么求答案。

$$\begin{aligned} 令V(r,c)&=1当且仅当ar>bc且xr<yc，否则为0\\ Ans &=\sum_{r=1}^n \sum_{c=1,ar>bc,xr<yc}^n \frac{r}{(r,c)}+\frac{c}{(r,c)}-2\\ &= \sum_{d=1}^n g(\frac{n}{d})-2*\sum_{r=1}^n\sum_{c=1}^n V(r,c) \\ g(n) &= \sum_{r=1}^n\sum_{c=1,(r,c)=1}^n (r+c)V(r,c)\\ &=\sum_{d=1}^n d*\mu(d)*\sum_{r=1}^{n/d} \sum_{c=1}^{n/d} (r+c)V(r,c)\\ 令g_0(n)&=\sum_{r=1}^{n} \sum_{c=1}^{n} (r+c)V(r,c) \\ g(n) &= \sum_{d=1}^n d*\mu(d)*g_0(\frac{n}{d}) \\ 令Value&=2*\sum_{r=1}^n\sum_{c=1}^n V(r,c)，这个显然可以用类欧几里德求\\ Ans &= \sum_{d=1}^n \sum_{p=1}^{\frac{n}{d}}p*\mu(p)*g_0(\frac{n}{dp}) -Value \\ Ans&=\sum_{d=1}^n \left(\sum_{p|d}p*\mu(p)\right)g_0(\frac{n}{d}) \\ 令h(n)&=\sum_{p|n} p*\mu(p)\\ Ans &= \sum_{d=1}^n h(d)g_0(\frac{n}{d}) \end{aligned}$$
不同的 $\frac{n}{d}$只有 $O(\sqrt{n})$种，对于某个 $g_0(n)$，把约束移下项，显然就可以用类欧几里德算法来做了，具体做法可以参考 这篇文章
把所有的 $g_0(n)$预处理出来的复杂度是 $O(\sqrt{n}\log n)$。
那么，只有能把 $\sum_{d=1}^n h(d)$的 $O(\sqrt{n})$个值处理出来，这题就做完了。
Google一发后可以发现， $h \times \phi = e$（这里是狄利克雷卷积），于是用杜教筛就好了。维护一下 $\mu,\phi,h$各自的前缀和就可以 $O(n^{\frac{2}{3}})$做完了。

因此这题的复杂度就是$O(T(\sqrt{n}\log n +n^{\frac{2}{3}}) + |S|)$，时限卡得有点紧，有些东西能预处理的需要尽量预处理。