結果

問題 No.1857 Gacha Addiction
ユーザー sasa8uyauya
提出日時 2025-02-24 03:04:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,947 ms / 6,000 ms
コード長 1,471 bytes
コンパイル時間 671 ms
コンパイル使用メモリ 82,908 KB
実行使用メモリ 135,624 KB
最終ジャッジ日時 2025-02-24 03:05:14
合計ジャッジ時間 60,791 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353
R = 3
Rinv = 332748118
W = [pow(R, (mod-1)>>i, mod) for i in range(24)]
Winv = [pow(Rinv, (mod-1)>>i, mod) for i in range(24)]


def fft(k, f):
	for l in range(k, 0, -1):
		d = 1<<l-1
		U = [1]
		for i in range(d):
			U.append(U[-1]*W[l]%mod)
		for i in range(1<<k-l):
			for j in range(d):
				s = i*2*d+j
				f[s], f[s+d] = (f[s]+f[s+d])%mod, U[j]*(f[s]-f[s+d])%mod


def fftinv(k, f):
	for l in range(1, k+1):
		d = 1<<l-1
		for i in range(1<<k-l):
			u = 1
			for j in range(i*2*d, (i*2+1)*d):
				f[j+d] *= u
				f[j], f[j+d] = (f[j]+f[j+d])%mod, (f[j]-f[j+d])%mod
				u *= Winv[l]
				u %= mod


def convolution(a, b):
	le = len(a)+len(b)-1
	k = le.bit_length()
	n = 1<<k
	a = a+[0]*(n-len(a))
	b = b+[0]*(n-len(b))
	fft(k, a)
	fft(k, b)
	for i in range(n):
		a[i] *= b[i]
		a[i] %= mod
	fftinv(k, a)
	ninv = pow(n, mod-2, mod)
	for i in range(le):
		a[i] *= ninv
		a[i] %= mod
	return a[:le]



n,S=map(int,input().split())
p=list(map(int,input().split()))
M=998244353
iS=pow(S,M-2,M)
N=n
fa=[1]
for i in range(1,N+1):
  fa+=[fa[-1]*i%M]
fb=[pow(fa[N],M-2,M)]
for i in reversed(range(1,N+1)):
  fb+=[fb[-1]*i%M]
fb.reverse()

def f(l,r):
  if l==r:
    return [1,p[l]*iS%M]
  m=(r-l+1)//2
  ql=f(l,l+m-1)
  qr=f(l+m,r)
  for i in range(len(ql)):
    ql[i]*=fb[i]
    ql[i]%=M
  for i in range(len(qr)):
    qr[i]*=fb[i]
    qr[i]%=M
  q=convolution(ql,qr)
  for i in range(len(q)):
    q[i]*=fa[i]
    q[i]%=M
  return q

q=f(0,n-1)
print(sum(q)%M)
0