結果
問題 | No.1766 Tatsujin Remix |
ユーザー | rlangevin |
提出日時 | 2023-02-20 23:10:45 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 127 ms / 2,000 ms |
コード長 | 1,395 bytes |
コンパイル時間 | 165 ms |
コンパイル使用メモリ | 82,404 KB |
実行使用メモリ | 94,152 KB |
最終ジャッジ日時 | 2024-07-21 13:15:20 |
合計ジャッジ時間 | 3,964 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 22 |
ソースコード
import sys readline = sys.stdin.readline N = int(readline()) mod = 998244353 S = [] for i in range(N): s = readline().rstrip() S.extend(list(s)) N *= 16 dp0 = [0] * N dp1 = [0] * N dp2 = [0] * N if S[0] == "d": dp1[0] = 1 elif S[0] == "k": dp2[0] = 1 else: dp0[0] = 1 dp1[0] = 1 dp2[0] = 1 for i in range(1, N): if S[i] == ".": if i % 2: dp0[i] += dp0[i - 1] + dp1[i - 1] + dp2[i - 1] dp1[i] += dp2[i - 1] + dp1[i - 1] dp2[i] += dp1[i - 1] + dp2[i - 1] else: dp0[i] += dp0[i - 1] dp1[i] += dp2[i - 1] + dp0[i - 1] + dp1[i - 1] dp2[i] += dp1[i - 1] + dp0[i - 1] + dp2[i - 1] elif S[i] == "d": if i % 2 == 0: dp1[i] += dp0[i - 1] + dp1[i - 1] + dp2[i - 1] if S[i - 1] == ".": dp1[i] -= dp1[i - 1] else: dp1[i] += dp1[i - 1] + dp2[i - 1] if S[i - 1] == ".": dp1[i] -= dp1[i - 1] else: if i % 2 == 0: dp2[i] += dp0[i - 1] + dp1[i - 1] + dp2[i - 1] if S[i - 1] == ".": dp2[i] -= dp2[i - 1] else: dp2[i] += dp1[i - 1] + dp2[i - 1] if S[i - 1] == ".": dp2[i] -= dp2[i - 1] dp0[i] %= mod dp1[i] %= mod dp2[i] %= mod print( (dp0[-1] + dp1[-1] + dp2[-1] ) % mod )