結果

問題 No.2833 Count Taiko Results
コンテスト
ユーザー 👑 p-adic
提出日時 2024-06-21 08:47:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 394 ms / 2,000 ms
コード長 336 bytes
コンパイル時間 181 ms
コンパイル使用メモリ 82,540 KB
実行使用メモリ 115,172 KB
最終ジャッジ日時 2024-06-21 10:09:13
合計ジャッジ時間 8,892 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 58
権限があれば一括ダウンロードができます

ソースコード

diff # 

R=range
J=lambda:map(int,input().split())
N,K=J()
A=list(J())+[0]
B=list(J())
a=0
t=[1]+[0]*N
P=998244353
for m in R(K-1,K+1):
	s=p=1
	for i in R(N):
		t[i+1]=s*B[i]%P;s=((s if i<m else s-p*t[i-m])*A[i]+t[i+1])%P;p=p*A[i]%P
		if i>=m:p=p*pow(A[i-m],-1,P)%P
	s=0;p=1
	for k in R(m+1):s+=t[N-k]*p;p=p*A[N-1-k]%P
	a+=[s,-s][m<K]
print(a%P)
0