結果

問題 No.2964 Obstruction Bingo
ユーザー hato336
提出日時 2024-11-16 16:09:04
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,226 bytes
コンパイル時間 364 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 528,896 KB
最終ジャッジ日時 2024-11-16 16:11:17
合計ジャッジ時間 112,147 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24 TLE * 24 MLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import collections
l,k = map(int,input().split())
s = list(input().rstrip())
t = list(input().rstrip())
for i in range(l):
    s[i] = ord(s[i]) - ord('a')
    t[i] = ord(t[i]) - ord('a')
a = list(map(int,input().split()))
mod = 998244353
inv = pow(sum(a),mod-2,mod)
su = sum(a)
dp = [[0 for i in range(k+2)] for j in range(k+2)]
dp[0][0] = 1
ans = [0,0]
for i in range(k+1):
    ndp = [[0 for wrfrf in range(k+2)] for  fwfwf in range(k+2)]
    for j in range(k+1):
        for m in range(k+1):
            if abs(j-m) == l:
                if j > m:
                    ans[0] += dp[j][m]
                    ans[0] %= mod
                else:
                    ans[1] += dp[j][m]
                    ans[1] %= mod
                continue
            if abs(j-m) > l:
                continue
            if s[j%l] != t[m%l]:
                ndp[j][m] += (dp[j][m] * (su - a[s[j%l]] - a[t[m%l]]) * inv) % mod
                ndp[j+1][m] += (dp[j][m] * a[s[j%l]] * inv) % mod
                ndp[j][m+1] += (dp[j][m] * a[t[m%l]] * inv) % mod
            else:
                ndp[j][m] += (dp[j][m] * (su - a[s[j%l]]) * inv) % mod
                ndp[j+1][m+1] += (dp[j][m] * a[s[j%l]] * inv) % mod
    dp = ndp
print(*ans)
0