結果

問題 No.1740 Alone 'a'
ユーザー H20H20
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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())
0