結果

問題 No.2993 冪乗乗 mod 冪乗
ユーザー 👑 p-adic
提出日時 2023-11-02 12:29:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 9,781 ms / 10,000 ms
コード長 1,041 bytes
コンパイル時間 390 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 82,432 KB
最終ジャッジ日時 2024-12-22 19:42:08
合計ジャッジ時間 56,202 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

I,R=input,range
L=1000
P=[]
C=[0]*L
for i in R(2,L):
	if C[i]<1:
		P+=[i]
		j=i*2
		while j<L:C[j],j=1,j+i
def S(p):
	global n,u,f,q,l,B
	o=p-1
	for j in R(n):
		while o%f[j]<1:o//=f[j]
	o,n=(p-1)//o,n+1
	f+=[p]
	e=v=a=0
	while B%p<1:e,B=e+N,B//p
	r=pow(p,e)
	q+=[r]
	d=p*r
	while d<2*e*N*r+r:d*=p
	m=t=(1-pow(M,o,d))%d
	i,s=[0,1],2
	if m==0:
		l+=[0]
	else:
		while t%p<1:t,v=t//p,v+1
		t=m
		if v<1:u=1
		else:
			for k in R(1,(2*e//v+1)*N+2):
				j=k
				while j%p<1:j=j//p
				while s<=j:
					i+=[0if s%p<1 else r-r//s*i[r%s]%r if i[r%s]else(r//s+1)*i[s-r%s]%r]
					s+=1
				a,t=a-t//(k//j)*i[j],t*m%d
			l+=[pow(o,-1,r)*a%r]
for t in R(int(I())):
	B,N,M=map(int,I().split())
	D=B**N
	n=u=0
	f,q,l=[],[],[]
	for p in P:
		if B%p<1:S(p)
	if B>1:S(B)
	if u:a=-1
	else:
		a,r=0,1
		for k in R(n):
			b,c=[r,q[k]],[[1,0],[0,1]]
			i=r<q[k]
			j=1-i
			while b[j]:
				d=b[i]//b[j]
				c[i][i]-=d*c[j][i]
				c[i][j]-=d*c[j][j]
				b[i]-=d*b[j]
				i,j=j,i
			g=b[i]
			a,r=(l[k]%g+l[k]//g*r*c[i][0]+a//g*q[k]*c[i][1])%D,r*q[k]//g
	print(a)
0