結果

問題 No.2833 Count Taiko Results
ユーザー 👑 p-adic
提出日時 2024-06-21 08:57:18
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,589 ms / 2,000 ms
コード長 336 bytes
コンパイル時間 167 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 41,012 KB
最終ジャッジ日時 2024-06-21 10:09:33
合計ジャッジ時間 27,694 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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