結果
問題 | No.2874 Gunegune Tree |
ユーザー | n0ma_ru |
提出日時 | 2024-09-21 15:05:20 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 53 ms / 2,000 ms |
コード長 | 1,561 bytes |
コンパイル時間 | 2,401 ms |
コンパイル使用メモリ | 208,780 KB |
実行使用メモリ | 20,436 KB |
最終ジャッジ日時 | 2024-09-21 15:05:26 |
合計ジャッジ時間 | 4,270 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 2 ms
5,376 KB |
testcase_03 | AC | 5 ms
5,376 KB |
testcase_04 | AC | 22 ms
10,476 KB |
testcase_05 | AC | 10 ms
5,860 KB |
testcase_06 | AC | 2 ms
5,376 KB |
testcase_07 | AC | 48 ms
19,296 KB |
testcase_08 | AC | 24 ms
10,844 KB |
testcase_09 | AC | 5 ms
5,376 KB |
testcase_10 | AC | 41 ms
16,872 KB |
testcase_11 | AC | 32 ms
14,048 KB |
testcase_12 | AC | 11 ms
6,712 KB |
testcase_13 | AC | 47 ms
19,252 KB |
testcase_14 | AC | 40 ms
16,440 KB |
testcase_15 | AC | 8 ms
5,596 KB |
testcase_16 | AC | 51 ms
20,092 KB |
testcase_17 | AC | 8 ms
5,376 KB |
testcase_18 | AC | 8 ms
5,376 KB |
testcase_19 | AC | 7 ms
5,376 KB |
testcase_20 | AC | 7 ms
5,376 KB |
testcase_21 | AC | 19 ms
8,800 KB |
testcase_22 | AC | 26 ms
11,304 KB |
testcase_23 | AC | 18 ms
8,280 KB |
testcase_24 | AC | 22 ms
9,916 KB |
testcase_25 | AC | 37 ms
14,800 KB |
testcase_26 | AC | 24 ms
10,184 KB |
testcase_27 | AC | 53 ms
19,692 KB |
testcase_28 | AC | 13 ms
6,912 KB |
testcase_29 | AC | 31 ms
12,856 KB |
testcase_30 | AC | 25 ms
10,808 KB |
testcase_31 | AC | 19 ms
8,904 KB |
testcase_32 | AC | 52 ms
20,436 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; #define ALL(v) v.begin(),v.end() #define dbg(x) cerr << #x << ": " << (x) << endl; ll modpow(ll a, ll n, ll mod) { a %= mod; ll ans = 1; while (n) { if (n & 1) ans = ans * a % mod; a = a * a % mod; n >>= 1; } return ans; } int N; int main() { cin >> N; ll mod = 998244353; vector exp(5, vector<ll>(N)), prob(5, vector<ll>(N)); vector<bitset<5>> next(5); next[0].set(0); next[0].set(1); next[1].set(0); next[1].set(1); next[1].set(2); next[2].set(1); next[2].set(2); next[2].set(3); next[3].set(2); next[3].set(3); next[3].set(4); next[4].set(3); next[4].set(4); vector<int> inv(6); for (int i = 1; i < 6; ++i) inv[i] = modpow(i, mod-2, mod); exp[1][0] = exp[2][0] = exp[3][0] = inv[5]; for (int d = 0; d < 5; ++d) prob[d][0] = inv[5]; for (int i = 1; i < N; ++i) { for (int d = 0; d < 5; ++d) { for (int to = 0; to < 5; ++to) { if (next[d].test(to)) { int grow = (1 <= to && to <= 3) * prob[d][i-1]; exp[to][i] += (exp[d][i-1] + grow) * inv[ next[d].count() ] % mod; exp[to][i] %= mod; prob[to][i] += prob[d][i-1] * inv[ next[d].count() ] % mod; prob[to][i] %= mod; } } } } ll ans = 0; for (int i = 0; i < 5; ++i) ans += exp[i][N-1]; ans %= mod; cout << ans << '\n'; }