結果
問題 | No.1740 Alone 'a' |
ユーザー | H20 |
提出日時 | 2021-11-12 22:40:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,331 ms / 2,000 ms |
コード長 | 635 bytes |
コンパイル時間 | 484 ms |
コンパイル使用メモリ | 82,368 KB |
実行使用メモリ | 134,744 KB |
最終ジャッジ日時 | 2024-11-25 20:20:00 |
合計ジャッジ時間 | 21,750 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 38 |
ソースコード
from itertools import product def count(): n = int(input()) a = input() mod = 998244353 #配列は末から dp=[[[0] * 3 for k in range(2)] for l in range(n+1)] dp[0][0][0] = 1 #条件に合わせてDP for i, less, cnta in product(range(n), (0,1), range(3)): max_d = 25 if less else int(ord(a[i])-97) for d in range(max_d+1): less_ = less or d < max_d cnta_ = min(2, cnta + int(d==0)) dp[i + 1][less_][cnta_] = (dp[i][less][cnta]+dp[i + 1][less_][cnta_])%mod #合致するものを合算 ret = dp[n][1][1] return ret print(count())