結果

問題 No.2964 Obstruction Bingo
コンテスト
ユーザー sasa8uyauya
提出日時 2024-11-16 16:52:37
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 742 bytes
コンパイル時間 602 ms
コンパイル使用メモリ 81,792 KB
実行使用メモリ 558,828 KB
最終ジャッジ日時 2024-11-16 16:55:11
合計ジャッジ時間 118,711 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17 TLE * 28 MLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

L,K=map(int,input().split())
s=list(input())
for i in range(L):
  s[i]=ord(s[i])-ord("a")
t=list(input())
for i in range(L):
  t[i]=ord(t[i])-ord("a")
a=list(map(int,input().split()))
M=998244353
A=sum(a)
iA=pow(A,M-2,M)
X=10**10
w1=0
w2=0
q=[[0]*(2*L+1) for i in range(L)]
q[0][0]=1
import gc
for _ in range(K):
  nq=[[0]*(2*L+1) for i in range(L)]
  for i in range(L):
    for l in range(-L+1,L):
      j=(i-l)%L
      if s[i]==t[j]:
        nq[(i+1)%L][l]+=q[i][l]*a[s[i]]*iA
        nq[i][l]+=q[i][l]*(A-a[s[i]])*iA
      else:
        nq[(i+1)%L][l+1]+=q[i][l]*a[s[i]]*iA
        nq[i][l-1]+=q[i][l]*a[t[j]]*iA
        nq[i][l]+=q[i][l]*(A-a[s[i]]-a[t[j]])*iA
  q=nq
  for i in range(L):
    w1+=q[i][L]
    w2+=q[i][-L]
print(w1%M,w2%M)
0