結果
問題 | No.2874 Gunegune Tree |
ユーザー |
![]() |
提出日時 | 2024-09-06 22:25:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 69 ms / 2,000 ms |
コード長 | 588 bytes |
コンパイル時間 | 381 ms |
コンパイル使用メモリ | 82,380 KB |
実行使用メモリ | 75,992 KB |
最終ジャッジ日時 | 2024-09-06 22:25:46 |
合計ジャッジ時間 | 2,562 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
N = int(input()) mod = 998244353 inv2 = pow(2, mod - 2, mod) inv3 = pow(3, mod - 2, mod) inv5 = pow(5, mod - 2, mod) ans = 0 pre = [inv5] * 5 for i in range(N): for j in range(1, 4): ans += pre[j] ans %= mod dp = [0, 0, 0, 0 ,0] dp[0] = (pre[0] * inv2 + pre[1] * inv3) % mod dp[1] = (pre[0] * inv2 + pre[1] * inv3 + pre[2] * inv3) % mod dp[2] = (pre[1] * inv3 + pre[2] * inv3 + pre[3] * inv3) % mod dp[3] = (pre[4] * inv2 + pre[3] * inv3 + pre[2] * inv3) % mod dp[4] = (pre[4] * inv2 + pre[3] * inv3) % mod dp, pre = pre, dp print(ans)