結果
問題 |
No.2898 Update Max
|
ユーザー |
![]() |
提出日時 | 2024-09-25 16:19:22 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,346 bytes |
コンパイル時間 | 845 ms |
コンパイル使用メモリ | 88,984 KB |
実行使用メモリ | 7,640 KB |
最終ジャッジ日時 | 2024-09-25 16:19:26 |
合計ジャッジ時間 | 3,111 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 WA * 1 |
other | AC * 16 WA * 12 |
ソースコード
#include <iostream> #include <vector> using namespace std; const int MAXN = 2e5 + 5, Mod = 998244353; bool vis[MAXN]; vector<int> se; int n, m, k, ans, a[MAXN], b[MAXN], fac[MAXN] = {1, 1}, inv[MAXN] = {0, 1}, facinv[MAXN] = {1, 1}; int A(int n, int m) { if (n < m) return 0; return 1ll * fac[n] * facinv[n - m] % Mod; } int qpow(int x, int k) { int ret = 1; for (; k; k >>= 1, x = 1ll * x * x % Mod) { if (k & 1) ret = 1ll * x * ret % Mod; } return ret; } int main() { ios::sync_with_stdio(0), cin.tie(0); cin >> n; for (int i = 1; i <= n; i++) { cin >> a[i]; if (a[i] > 0) { vis[a[i]] = 1; } } for (int i = 2; i <= n; i++) { fac[i] = 1ll * fac[i - 1] * i % Mod; inv[i] = 1ll * (Mod - Mod / i) * inv[Mod % i] % Mod; facinv[i] = 1ll * facinv[i - 1] * inv[i] % Mod; if (!vis[i]) se.push_back(i); } m = se.size(); for (int i = 1, x = 0, c = 0; i <= n; i++) { if (a[i] > 0) { if (a[i] > x) { int y = lower_bound(se.begin(), se.end(), a[i]) - se.begin(); ans = (ans + 1ll * A(y, c) * fac[m - c]) % Mod; } } else { int y = lower_bound(se.begin(), se.end(), x) - se.begin(); c++, ans = (ans + 1ll * (A(m, c) - A(y, c) + Mod) * qpow(c, Mod - 2) % Mod * fac[m - c]) % Mod; } x = max(x, a[i]); } cout << ans; return 0; }