結果

問題 No.2131 Concon Substrings (COuNt Version)
コンテスト
ユーザー roaris
提出日時 2022-11-25 22:06:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 579 ms / 2,000 ms
コード長 663 bytes
コンパイル時間 218 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 97,912 KB
最終ジャッジ日時 2024-10-02 04:37:53
合計ジャッジ時間 5,380 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = sys.stdin.readline
from collections import *

N = int(input())
dp = [[0]*3 for _ in range(N+1)]
dp[0][0] = 1
MOD = 998244353

for _ in range(N):
    ndp = [[0]*3 for _ in range(N+1)]
    
    for i in range(N+1):
        ndp[i][0] += 25*dp[i][0]%MOD
        ndp[i][0] %= MOD
        ndp[i][1] += (25*dp[i][1]+dp[i][0])%MOD
        ndp[i][1] %= MOD
        ndp[i][2] += (25*dp[i][2]+dp[i][1])%MOD
        ndp[i][2] %= MOD
        
        if i+1<=N:
            ndp[i+1][0] += dp[i][2]
            ndp[i+1][0] %= MOD
    
    dp = ndp

ans = 0

for i in range(N+1):
    for j in range(3):
        ans += i*dp[i][j]
        ans %= MOD

print(ans)
0