結果

問題 No.2964 Obstruction Bingo
ユーザー hato336hato336
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
59,008 KB
testcase_01 AC 58 ms
71,340 KB
testcase_02 AC 107 ms
82,612 KB
testcase_03 AC 52 ms
334,136 KB
testcase_04 AC 51 ms
68,224 KB
testcase_05 AC 897 ms
435,408 KB
testcase_06 AC 2,072 ms
263,068 KB
testcase_07 AC 326 ms
354,424 KB
testcase_08 AC 1,960 ms
255,016 KB
testcase_09 AC 1,143 ms
457,440 KB
testcase_10 TLE -
testcase_11 AC 671 ms
359,880 KB
testcase_12 TLE -
testcase_13 AC 1,449 ms
465,144 KB
testcase_14 AC 2,119 ms
274,816 KB
testcase_15 AC 400 ms
363,332 KB
testcase_16 AC 659 ms
104,272 KB
testcase_17 AC 477 ms
359,268 KB
testcase_18 AC 1,052 ms
144,972 KB
testcase_19 AC 602 ms
362,340 KB
testcase_20 AC 2,229 ms
271,632 KB
testcase_21 MLE -
testcase_22 AC 343 ms
86,208 KB
testcase_23 AC 199 ms
349,024 KB
testcase_24 AC 166 ms
83,744 KB
testcase_25 TLE -
testcase_26 TLE -
testcase_27 TLE -
testcase_28 TLE -
testcase_29 TLE -
testcase_30 TLE -
testcase_31 TLE -
testcase_32 TLE -
testcase_33 TLE -
testcase_34 TLE -
testcase_35 TLE -
testcase_36 TLE -
testcase_37 TLE -
testcase_38 TLE -
testcase_39 TLE -
testcase_40 TLE -
testcase_41 TLE -
testcase_42 TLE -
testcase_43 TLE -
testcase_44 TLE -
testcase_45 AC 1,947 ms
272,952 KB
testcase_46 AC 1,408 ms
271,232 KB
testcase_47 AC 1,417 ms
273,792 KB
testcase_48 AC 1,682 ms
268,996 KB
testcase_49 AC 2,103 ms
271,308 KB
testcase_50 TLE -
testcase_51 TLE -
権限があれば一括ダウンロードができます

ソースコード

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