洛谷 P3811 【模板】模意义下的乘法逆元

对于乘法逆元，我想大家再熟悉不过了，如过不知道的小伙伴们可以看乘法逆元来学习一下。

题目描述

给定 𝑛,𝑝 求 1∼𝑛 中所有整数在模 𝑝 意义下的乘法逆元。

这里 𝑎 模 𝑝 的乘法逆元定义为 𝑎𝑥≡1(mod 𝑝) 的解。

输入格式

一行两个正整数 𝑛,𝑝。

输出格式

输出 𝑛 行，第 𝑖 行表示 𝑖 在模 𝑝 下的乘法逆元。

输入输出样例

输入 #1

10 13

输出 #1

1
7
9
10
8
11
2
5
3
4

说明/提示

1≤𝑛≤3×106，𝑛<𝑝<20000528。

输入保证 𝑝 为质数。

对于这道是一道乘法逆元的模板题，我们要求1~n之间在mod p下的乘法逆元；

根据题目要求我们可以直接写出暴力代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,p,fac[3000005],inv[3000005];
int binpow(int a,int b)
{
    int res=1;
    while(b)
    {
        if(b&1)
        {
            res*=a;
            res%=p;
        }
        a*=a;
        a%=p;
        b>>=1;
    }
    return res;
}
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin>>n>>p;
    for(int i=1;i<=n;++i)
    {
        cout<<binpow(i%p,p-2)%p<<"\n";
    }
​
    return 0;
}

这段代码，虽然可以实现，但这道题卡我们的O(nlogn)时间复杂度，只有500ms我们可以考虑每次O(1)的得到答案，把时间复杂度降为O(n)。（详见乘法逆元 - OI Wiki）

所以我们可以写出O(n),时间复杂度的代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,p,fac[3000005],inv[3000005];
int binpow(int a,int b)
{
    int res=1;
    while(b)
    {
        if(b&1)
        {
            res*=a;
            res%=p;
        }
        a*=a;
        a%=p;
        b>>=1;
    }
    return res;
}
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin>>n>>p;
    fac[0]=1;
    for(int i=1;i<=n;++i)
    {
        fac[i]=fac[i-1]*i;
        fac[i]%=p;
    }
    inv[n]=binpow(fac[n],p-2);
    for(int i=n-1;i>=1;--i)
    {
        inv[i]=inv[i+1]*(i+1)%p;
    }
    for(int i=1;i<=n;++i)
    {
        cout<<fac[i-1]*inv[i]%p<<"\n";
    }
​
    return 0;
}

好小伙伴们你们都学废了么？

如果感兴趣的小伙伴们可以看看洛谷 P5431 【模板】模意义下的乘法逆元 2（【模板】模意义下的乘法逆元 2 - 洛谷）