結果

問題 No.2578 Jewelry Store
ユーザー 👑 p-adic
提出日時 2023-12-07 09:49:07
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 776 bytes
コンパイル時間 474 ms
コンパイル使用メモリ 82,472 KB
実行使用メモリ 97,880 KB
最終ジャッジ日時 2024-09-27 02:01:12
合計ジャッジ時間 28,342 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19 WA * 8 TLE * 1 -- * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

R=range
I=lambda:map(int,input().split())
T,m=I()
M=[m]
P=set()
def gcd(a,b):
	while a:a,b=b%a,a
	return b
while M:
	n=M.pop()
	if n<99:
		for i in R(2,n+1):
			if n%i<1:
				P.add(i)
				while n%i<1:n//=i
	else:
		for t in R(int(n**0.25)*2+9):
			x,d=int(n**0.5)+1,1
			y=(x*x+1)%n
			while d<2:x,y,d=(x*x+1)%n,(y**4+2*y*y+2)%n,gcd(abs(x-y),n)
			if d<n:
				M+=[d,n//d]
				break
		else:P.add(n)
P=list(P)
L=len(P)
Q=998244353
for t in R(T):
	N,B,C,D=I()
	A=list(I())
	b=[0]*N
	for i in R(L):
		n=m
		while n%P[i]<1:n//=P[i]
		for j in R(N):b[j]|=(A[j]%(m//n)>0)*1<<i
	W=[]
	for a in A:
		W+=[1+B*(m%a<1)]
		B=(C*B+D)%Q
	v=0
	for d in R(1<<L):
		w,S=1,[i for i in R(L)if d>>i&1]
		for j in R(N):w*=[1,W[j]][all(b[j]>>i&1for i in S)]
		v+=[w,-w][len(S)%2]
	print((v-(m<2))%Q)
0