結果

問題 No.2964 Obstruction Bingo
コンテスト
ユーザー sasa8uyauya
提出日時 2024-11-16 16:26:58
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 778 bytes
コンパイル時間 526 ms
コンパイル使用メモリ 82,016 KB
実行使用メモリ 193,532 KB
最終ジャッジ日時 2024-11-16 16:29:29
合計ジャッジ時間 147,889 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 41
権限があれば一括ダウンロードができます

ソースコード

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 j in range(L)] for i in range(L)]
q[0][0][0]=1
for _ in range(K):
  nq=[[[0]*(2*L+1) for j in range(L)] for i in range(L)]
  for i in range(L):
    for j in range(L):
      for l in range(-L+1,L):
        for c in range(26):
          nq[(i+(c==s[i]))%L][(j+(c==t[j]))%L][l+(c==s[i])-(c==t[j])]+=q[i][j][l]*a[c]*iA
          nq[(i+(c==s[i]))%L][(j+(c==t[j]))%L][l+(c==s[i])-(c==t[j])]%=M
  q=nq
  for i in range(L):
    for j in range(L):
      w1+=q[i][j][L]
      w1%=M
      w2+=q[i][j][-L]
      w2%=M
print(w1,w2)
0