結果

問題 No.2131 Concon Substrings (COuNt Version)
ユーザー tamato
提出日時 2022-11-25 21:29:47
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 524 bytes
コンパイル時間 152 ms
コンパイル使用メモリ 81,588 KB
実行使用メモリ 60,752 KB
最終ジャッジ日時 2024-10-02 04:02:19
合計ジャッジ時間 1,811 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 998244353


def main():
    import sys
    input = sys.stdin.readline

    N = int(input())

    dp = [0] * 3
    dp[0] = 1
    ans = 0
    for j in range(N):
        dp_new = [0] * 3
        for i in range(3):
            dp_new[i] = (dp_new[i] + (dp[i] * 25) % mod) % mod
            dp_new[(i + 1) % 3] = (dp_new[(i + 1) % 3] + dp[i]) % mod
            if i == 2:
                ans = (ans + (dp[i] * pow(26, N - j - 1, mod)) % mod) % mod
        dp = dp_new
    print(ans)


if __name__ == '__main__':
    main()
0