結果
問題 | 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 |
ソースコード
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)