【GDKOI2016】小学生数学题（附带了乘法取模黑科技）

题目描述

给定$n, p, k$，其中$p$是质数。
求$\sum_{i=1}^n \frac{1}{i} \pmod {p^k}$
题目保证答案是$\frac{P}{Q}$的形式，且$Q^{-1}$，即$Q$对$p^k$的逆元存在。

$np^k\leq10^{18}$
$p\leq 10^5$

---

分析

先考虑所有变量的范围
$k\leq 70$
$n\leq 10^{18}$

类似于求$n! \mod p$的思路。

$\sum_{i=1}^n \frac{1}{i}=\frac{1}{p}\sum_{i=1}^{\lfloor \frac{n}{p}\rfloor}\frac{1}{i}+\sum_{0<a<p, a+bp\leq n}\frac{1}{a+bp}$

那么这个式子的计算就分成了两个部分。

part 1

$\frac{1}{p}\sum_{i=1}^{\lfloor \frac{n}{p}\rfloor}\frac{1}{i} \pmod {p^k}$

假如我们可以处理好$\frac{1}{p}$的问题，那么剩下的递归计算就可以了。
考虑计算

$\sum_{i=1}^{\lfloor \frac{n}{p}\rfloor}\frac{1}{i} \pmod {p^{k+1}}$

那么由于题目保证了答案逆元存在，所以

$\frac{1}{p}\sum_{i=1}^{\lfloor \frac{n}{p}\rfloor}\frac{1}{i}=\frac{\sum_{i=1}^{\lfloor \frac{n}{p}\rfloor}\frac{1}{i} \mod {p^{k+1}}}{p} \pmod {p^k}$

分子必定含有$p$的因子，具体证明的话，

$a = b\pmod c \Leftrightarrow ap=bp\pmod {cp}$

结合这个式子倒推一下就可以得到了。

part 2

$\sum_{0<a<p, a+bp\leq n}\frac{1}{a+bp}$

现在要处理的就是快速地计算$\sum(a+bp)^{-1}$
考虑按$a$分组，推导

$原式=\sum_{b=0}^{\lfloor\frac{n-a}{p}\rfloor}\frac{1}{a+bp}=\frac{\sum_{b=0}^{\lfloor\frac{n-a}{p}\rfloor}\frac{1}{1+\frac{bp}{a}}}{a}\pmod {p^k}$

考虑$\frac{1}{1-x}=x^0+x^1+x^2+\cdots$
那么

$原式=\frac{\sum_{b=0}^{\lfloor\frac{n-a}{p}\rfloor}\sum_{k=0}^{+\infty}(-\frac{bp}{a})^k}{a}=\frac{\sum_{b=0}^{\lfloor\frac{n-a}{p}\rfloor}\sum_{i=0}^{+\infty}(-\frac{bp}{a})^i}{a}\pmod {p^k}$

由于括号内那项含$p$，因此当$i\geq k$时，和式里面的项就为$0$了。
考虑交换枚举顺序，则有

$原式=\frac{\sum_{i=0}^{k-1}(-\frac{p}{a})^i\sum_{b=0}^{\lfloor\frac{n-a}{p}\rfloor}b^i}{a}\pmod {p^k}$

问题成功地转化成了自然数幂求和问题，用矩阵乘法处理一下就可以了。

考虑时间复杂度。

$T(n, k)=T(\frac{n}{p}, k+1)+k^3+kP$

因此总的时间复杂度上界约为$O(k^3log_p^n+kPlog_p^n)$

---

备注

注意中间可能要有64位整数相乘后取模的过程。
用黑科技就可以跑得飞快了。

附一下黑科技的代码

LL mult( LL A, LL B, LL Mo )
{
    LL temp = ( ( LL ) ( ( db ) A*B/Mo+1e-6 ) * Mo );
    return A*B - temp;
}