結果

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

ソースコード

diff # 

R=range
I=lambda:map(int,input().split())
T,m=I()
M=[m]*(m>1)
P=[]
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):
			if n%i<1:
				P+=[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]
				while n%d<1:n//=d
				M+=[n]*(n>0)
				break
		else:P+=[n]
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