結果
問題 |
No.2131 Concon Substrings (COuNt Version)
|
ユーザー |
|
提出日時 | 2022-11-29 12:52:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 245 ms / 2,000 ms |
コード長 | 586 bytes |
コンパイル時間 | 408 ms |
コンパイル使用メモリ | 82,172 KB |
実行使用メモリ | 76,800 KB |
最終ジャッジ日時 | 2024-10-06 23:42:19 |
合計ジャッジ時間 | 4,047 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 16 |
ソースコード
N = int(input()) P = 998244353 ans = 0 inf = 1000 dp = [[0] * (inf + 1) for _ in range(3)] dp[0][0] = 1 for _ in range(N): nx = [[0] * (inf + 1) for _ in range(3)] for i in range(inf + 1): nx[0][i] += dp[0][i] * 25 nx[1][i] += dp[1][i] * 25 + dp[0][i] nx[2][i] += dp[2][i] * 25 + dp[1][i] if i < inf: nx[0][i + 1] += dp[2][i] for i in range(inf + 1): for j in range(3): nx[j][i] %= P dp = nx ans = 0 for i in range(inf + 1): for j in range(3): ans += dp[j][i] * i ans %= P print(ans)