[洛谷] P5431 【模板】乘法逆元2

题目

题目描述

给定$N$个正整数$a_i$，求它们在模$P$意义下的乘法逆元。
由于输出太多不好，所以将会给定常数$K$，你要输出的答案为：
$$\sum _{i=1}^n\frac{k^i}{a_i}$$
当然要对$P$取模。

题目解析

我们先把式子中的 $a_i$ 替换成它的逆元，变成，${\sum }_{i=1}^n{k}^i\times a_i^{-1}$

然后再求出 $a_i$ 的前缀积，$s[i]=s[i-1]\times a_i$

设 $t_i$ 为 $a_i^{-1}$ 的前缀积，我们可以发现可以求出$a_i^{-1}=t_i\times s_{i-1}$

然后要求 $t_i$，根据费马小定理 $a^{p-1}\equiv 1$ $(mod$ $p)$，可以得出，$t_n=s_n^{p-2}$

可以推出$t_i=t_{i+1}\times a_{i+1}$，从而得出数组 $t$

还可以快读优化

原式=$a_1^{-1}{k}^1+a_2^{-1}{k}^2+...+a_{n-1}^{-1}{k}^{n-1}+a_n^{-1}{k}^n$ $=$ $(k(a_1^{-1}+k(a_2^{-1})+...+k(a_{n-1}^{-1}+k{a}_n^{-1})))$

代码

#include<bits/stdc++.h>
#define ll long long
#define N 5000005
using namespace std;
ll n,p,k,ans,x;
ll a[N],s[N],t[N];
ll read()
{
	ll ans=0;
	char c=getchar();
	while(c<'0'||c>'9') c=getchar();
	while(c>='0'&&c<='9') ans=ans*10+c-48,c=getchar();
	return ans;
}
ll qp(ll a,ll b)
{
	ll t=1;
	for(;b;b>>=1,(a*=a)%=p)
	 if(b&1) (t*=a)%=p;
	return t;
}
int main()
{
	n=read();p=read();k=read();
	x=k;s[0]=1;
	for(int i=1;i<=n;i++)
	 a[i]=read(),s[i]=s[i-1]*a[i]%p;
	t[n]=qp(s[n],p-2);
	for(int i=n-1;i>=1;i--)
	 t[i]=t[i+1]*a[i+1]%p;
	for(int i=n;i>=1;--i)
     ans=((t[i]*s[i-1])%p+ans)*k%p;
	printf("%lld",ans);
}