扩展Lucas定理

本介绍主要是在学习后写下自己的理解，故以转载形式发出。

题意： 给定$n,m,p$，求$C_n^m\mathrm{m}\mathrm{o}\mathrm{d}p$，其中$1\le m\le n\le 10^{18},2\le p\le 10^6$。
题解：

• 目标：求$C_n^m\mathrm{m}\mathrm{o}\mathrm{d}p$
• 对$p$分解质因数：$p=p_1^{{\alpha }_1}{p}_2^{{\alpha }_2}{p}_3^{{\alpha }_3}\dots$
• 联想到中国剩余定理，$p_k^{{\alpha }_k}$是互质的模数，设$x=C_n^m\mathrm{m}\mathrm{o}\mathrm{d}p$，所以有：
  $\{\begin{array}{l}{C}_n^m(\mathrm{m}\mathrm{o}\mathrm{d}{p}_1^{{\alpha }_1})\\ C_n^m(\mathrm{m}\mathrm{o}\mathrm{d}{p}_2^{{\alpha }_2})\\ ...\end{array}$

答案$x$就是合并所有同余式得到的答案。

• 问题转换为了：

  $C_n^m\mathrm{m}\mathrm{o}\mathrm{d}{P}^k,[p\in primes]$

  即：

  $\frac{n!}{m!(n-m)!}\mathrm{m}\mathrm{o}\mathrm{d}{P}^k$

  那么问题就是无法确定$n!$中的$P$是否对取模$P^k$有影响，以及如何求出$m!,(n-m)!$的逆元。由于$a$对$p$存在逆元的条件是$gcd(a,p)=1$
  所以先将$P$从中提出：

  $\frac{\frac{n!}{P^x}}{\frac{m!}{P^y}\frac{(n-m)!}{P^z}}{P}^{x-y-z}\mathrm{m}\mathrm{o}\mathrm{d}{P}^k$

• 显然求出$\frac{n!}{P^x}$即可。

• 对$n!$变形：

  $n!=1\cdot 2\cdot 3\cdot ...\cdot n=(P\cdot 2P\cdot 3P...)(1\cdot 2...)$

  $=P^{\lfloor \frac{n}{P}\rfloor }(1\cdot 2\cdot ...\cdot \lfloor \frac{n}{P}\rfloor )(1\cdot 2...)$

  $=P^{\lfloor \frac{n}{P}\rfloor }(\lfloor \frac{n}{P}\rfloor )!\underset{i=1,(imodP\ne 0)}{\prod ^n}i$

  显然，后面这个连乘式是具有循环节的，且最后会有一段不完整的部分，这个部分长度也可能为$0$。
  即：

  $P^{\lfloor \frac{n}{P}\rfloor }(\lfloor \frac{n}{P}\rfloor )!(\underset{i=1,(imodP\ne 0)}{\prod ^{P^k}}i{)}^{\lfloor \frac{n}{P^k}\rfloor }(\underset{i=P^k\lfloor \frac{n}{P^k}\rfloor ,(imodP\ne 0)}{\prod ^n}i)$

  这里取$P^k$是因为$P^k$是这里的模数。

  我们要求的是$\frac{n!}{P^x}$，注意这里的$(\lfloor \frac{n}{P}\rfloor )!$也可能有$P$，故定义函数：

  $f(n)=\frac{n!}{P^x}$

  故有：

  $f(n)=f(\lfloor \frac{n}{P})\rfloor (\underset{i=1,(imodP\ne 0)}{\prod ^{P^k}}i{)}^{\lfloor \frac{n}{P^k}\rfloor }(\underset{i=P^k\lfloor \frac{n}{P^k}\rfloor ,(imodP\ne 0)}{\prod ^n}i)$

  这样求解$f(n)$的时间复杂度为$O({\log }_{P}n)$
  考虑下边界即$f(0)=\frac{0!}{P^0}=1$
  此时我们要求的式子为：

  $\frac{\frac{n!}{P^x}}{\frac{m!}{P^y}\frac{(n-m)!}{P^z}}{P}^{x-y-z}\mathrm{m}\mathrm{o}\mathrm{d}{P}^k$

  $=\frac{f(n)}{f(m)f(n-m)}{P}^{x-y-z}\mathrm{m}\mathrm{o}\mathrm{d}{P}^k$

  此时$f(m),f(n-m)$都与$P^k$互质，所以用$exgcd$求逆元即可。
  那么此时问题还剩$P^{x-y-z}$的求解。
  对于该式：

  $P^{\lfloor \frac{n}{P}\rfloor }(\lfloor \frac{n}{P}\rfloor )!(\underset{i=1,(imodP\ne 0)}{\prod ^{P^k}}i{)}^{\lfloor \frac{n}{P^k}\rfloor }(\underset{i=P^k\lfloor \frac{n}{P^k}\rfloor ,(imodP\ne 0)}{\prod ^n}i)$

  幂存在于$P^{\lfloor \frac{n}{P}\rfloor }$和$(\lfloor \frac{n}{P}\rfloor )!$中：
  所以设

  $g(n)=\lfloor \frac{n}{P}\rfloor +g(\lfloor \frac{n}{P}\rfloor )$

  时间复杂度仍为$O({\log }_{P}n)$
  考虑下边界为$g(0)=0$
  最终式为：

  $\frac{\frac{n!}{P^x}}{\frac{m!}{P^y}\frac{(n-m)!}{P^z}}{P}^{g(n)-g(m)-g(n-m)}\mathrm{m}\mathrm{o}\mathrm{d}{P}^k$

  最后再用中国剩余定理合并即可。