O(n)递推求乘法逆元

题目链接

P3811 【模板】乘法逆元


题目描述

给定 \(n,p\), 求 \(1\sim n\) 中所有整数在模 \(p\) 意义下的乘法逆元。

输入格式

一行两个正整数 \(n,p\)。

输出格式

输出 \(n\) 行,第 \(i\) 行表示 \(i\) 在模 \(p\) 下的乘法逆元。

输入

10 13

输出

1
7
9
10
8
11
2
5
3
4

说明/提示

\(1 \leq n \leq 3 \times 10 ^ 6, n < p < 20000528\)
输入保证 \(p\) 为质数。

解题思路

  1. \(1^{−1}≡1(mod\,p)\)
  2. 设 \(p=k\times i +r,(1<r<i<p)\), \(k\) 是 \(p / i\) 的商,\(r\) 是余数,则:

\[k\times i+r≡0(mod\,p) \]

然后乘上\(i^{-1} ,r^{-1}\) 就可以得到:

\[k*r^{-1}+i^{-1}\equiv 0 \pmod p \]

\[i^{-1}\equiv -k*r^{-1} \pmod p \]

\[i^{-1}\equiv -\lfloor \frac{p}{i} \rfloor*(p \bmod i)^{-1} \pmod p \]

可以发现,这是个递推式~

  • 时间复杂度:\(O(n)\)

代码

#include<bits/stdc++.h>
using namespace std;
const int N=3e6+10;
int inv[N];
int n,p;
int main()
{
    scanf("%d%d",&n,&p);
    inv[1]=1;
    puts("1");
    for(int i=2;i<=n;i++)
    {
        inv[i]=1ll*(-p/i+p)*inv[p%i]%p;
        printf("%d\n",inv[i]);
    }
    return 0;
}
上一篇:同余3:解高次同余方程的BSGS算法及其扩展学习笔记


下一篇:Codeforces Round #548 div2 C