結果

問題 No.2993 冪乗乗 mod 冪乗
ユーザー 👑 p-adic
提出日時 2023-11-02 16:38:47
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,102 bytes
コンパイル時間 523 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 27,536 KB
最終ジャッジ日時 2024-12-17 23:36:53
合計ジャッジ時間 201,460 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 15 TLE * 15
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.py:39: SyntaxWarning: invalid decimal literal
  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]

ソースコード

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]
	c=0
	while p**c<e*N+1:c+=1
	d=p*r
	while d/r<e*N+c:d*=p
	m=t=1-pow(M,o)
	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,(e//v+1)*N+c+1):
				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]%r,t*m
			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