P5431 【模板】乘法逆元2

最新推荐文章于 2024-08-01 16:50:47 发布

crazy morning

最新推荐文章于 2024-08-01 16:50:47 发布

阅读量140

点赞数 1

分类专栏：数论

本文链接：https://blog.csdn.net/weixin_49843717/article/details/116834773

版权

线性递推逆元扩展欧几里得算法模运算数值计算

关键词由CSDN通过智能技术生成

数论专栏收录该内容

27 篇文章 0 订阅

订阅专栏

题目

思路

前缀积线性求逆元即可。
设要求 $a_1,a_2,a_3,…,a_n$ 的逆元。
那么设 $s_i=s_{i-1}*a_i$ ，我们求 $s_n$ 的逆元，则 $1\over{s_i}$ 可线性递推，进一步可得出 $1\over{a_i}$
code：

#include<iostream>
#include<algorithm>
#include<string>
#include<cmath>
#include<cstdio>
using namespace std;
long long n,p,k,a[5000006][3],o,s;
inline long long read()
{
	long long ans=0;
	char u=getchar();
	while (u>'9'||u<'0') u=getchar();
	while (u<='9'&&u>='0') ans=ans*10+u-'0',u=getchar();
	return ans;
}
pair<long long,long long> exgcd(long long a,long long b)
{
	if (b==0) return pair<long long,long long> (1,0);
	pair<long long,long long> tmp=exgcd(b,a%b);
	return pair<long long,long long> (tmp.second,tmp.first-a/b*tmp.second);
}
long long ff(long long x)
{
	pair<long long,long long> tmp=exgcd(x,p);
	return (tmp.first%p+p)%p;
}
int main()
{
	n=read(),p=read(),k=read();
	o=k;
	a[0][1]=a[0][2]=a[n+1][1]=a[n+1][2]=1;
	for (long long i=1;i<=n;i++)
	{
		a[i][0]=read();
		a[i][1]=a[i-1][1]*a[i][0]%p;
	}
	for (long long i=n;i>=1;i--) a[i][2]=a[i][0]*a[i+1][2]%p;
	for (long long i=1;i<=n;i++)
	{
		s=(s+o*a[i-1][1]%p*a[i+1][2]%p)%p;
		o=o*k%p;
	}
	printf("%lld",s*ff(a[1][2])%p);
    return 0;
}