【洛谷】P3811 【模板】乘法逆元

题目地址：

https://www.luogu.com.cn/problem/P3811

题目描述：
给定$n,p$求$1\sim n$中所有整数在模$p$意义下的乘法逆元。这里$a$模$p$的乘法逆元定义为$ax\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}p)$的解。

输入格式：
一行两个正整数$n,p$。

输出格式：
输出$n$行，第$i$行表示$i$在模$p$下的乘法逆元。

数据范围：
$1\le n\le 3\times 10^6,n<p<20000528$
输入保证$p$为质数。

单个数求模某个素数的逆元可以用费马小定理转化为快速幂来求。本题要求$1\sim n$每个数的模$p$逆元，这么做时间是$O(n\log n)$会超时，从而需要别的办法。设$f[i]$为$i$模$p$的逆元，则$f[1]=1$。考虑$f[x],x>1$，设$p=kx+b,k=\lfloor \frac{p}{x}\rfloor$，那么$$kx+b\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}p)$$由于$p$是素数，所以$$\begin{gathered} k{b}^{-1}+x^{-1}\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}p) \\ {x}^{-1}\equiv -\lfloor \frac{p}{x}\rfloor b^{-1}\equiv (p-\lfloor \frac{p}{x}\rfloor )(p\mathrm{m}\mathrm{o}\mathrm{d}x{)}^{-1}(\mathrm{m}\mathrm{o}\mathrm{d}p) \end{gathered}$$而$p\mathrm{m}\mathrm{o}\mathrm{d}x<x$，所以有$$f[x]=(p-\lfloor \frac{p}{x}\rfloor )f[p\mathrm{m}\mathrm{o}\mathrm{d}x]$$从而可以递推。代码如下：

#include <iostream>
using namespace std;

const int N = 3e6 + 10;
long inv[N];

int main() {
  int n, p;
  scanf("%d%d", &n, &p);
  inv[1] = 1;
  puts("1");
  for (int i = 2; i <= n; i++)
    printf("%ld\n", inv[i] = (long)p - (p / i) * inv[p % i] % p);
}

时空复杂度$O(n)$。