結果

問題 No.2964 Obstruction Bingo
ユーザー ニックネーム
提出日時 2024-11-16 17:36:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,108 ms / 2,468 ms
コード長 1,071 bytes
コンパイル時間 758 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 80,668 KB
最終ジャッジ日時 2024-11-16 17:37:40
合計ジャッジ時間 44,050 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #

l,k = map(int,input().split())
s = [ord(v)-97 for v in input()]
t = [ord(v)-97 for v in input()]
a = list(map(int,input().split()))
dp = [[0]*2*l for _ in range(2*l)]
dp[0][0] = 1; x = y = 0; z = 998244353
n = sum(a); d = pow(n,-1,z)
for _ in range(k):
    eq = [[0]*2*l for _ in range(2*l)]
    for i in range(2*l):
        for j in range(2*l):
            if i>=l<=j or abs(i-j)>=l or dp[i][j]==0: continue
            for v,c in enumerate(a):
                p = dp[i][j]*c%z*d%z
                if s[i%l]==t[j%l]==v:
                    if i+1>=l<=j+1: eq[(i+1)%l][(j+1)%l] += p
                    else: eq[i+1][j+1] += p
                elif s[i%l]==v:
                    if i+1-j==l: x += p
                    elif i+1>=l<=j: eq[(i+1)%l][j%l] += p
                    else: eq[i+1][j] += p
                elif t[j%l]==v:
                    if j+1-i==l: y += p
                    elif i>=l<=j+1: eq[i%l][(j+1)%l] += p
                    else: eq[i][j+1] += p
                else: eq[i][j] += p
    dp = [[eqij%z for eqij in eqi] for eqi in eq]
print(x%z,y%z)
0