結果

問題 No.1740 Alone 'a'
コンテスト
ユーザー ああいい
提出日時 2021-12-31 14:51:42
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,270 ms / 2,000 ms
コード長 977 bytes
コンパイル時間 302 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 77,056 KB
最終ジャッジ日時 2024-10-08 12:03:00
合計ジャッジ時間 22,321 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
S = input()
A = [ord(s) - ord('a') for s in S]
P = 998244353

dp = [[[0] * 2 for _ in range(2)] for _ in range(N)]

if A[0] > 0:
    dp[0][0][0] = A[0] - 1
    dp[0][0][1] = 1
    dp[0][1][0] = 1
    dp[0][1][1] = 0
else:
    dp[0][0][0] = 0
    dp[0][0][1] = 0
    dp[0][1][0] = 0
    dp[0][1][1] = 1
for i in range(N-1):
    dp[i+1][1][0] =(dp[i+1][1][0] + dp[i][0][0]) % P
    dp[i+1][0][0] =(dp[i+1][0][0] + dp[i][0][0] * 25) % P

    if A[i+1] == 0:
        dp[i+1][1][1] =(dp[i+1][1][1] +  dp[i][0][1]) % P
    else:
        dp[i+1][0][1] =(dp[i+1][0][1] + dp[i][0][1]) % P
        dp[i+1][0][0] =(dp[i+1][0][0] + dp[i][0][1] * (A[i+1] - 1)) % P
        dp[i+1][1][0] =(dp[i+1][1][0] + dp[i][0][1]) % P
    dp[i+1][1][0] =(dp[i+1][1][0] +  dp[i][1][0] * 25) % P
    if A[i+1] > 0:
        dp[i+1][1][1] =(dp[i+1][1][1] + dp[i][1][1]) % P
        dp[i+1][1][0] =(dp[i+1][1][0] + dp[i][1][1] * (A[i+1] - 1)) % P
ans = dp[N-1][1][0]
print(ans % P)
0