結果
問題 | No.2898 Update Max |
ユーザー | vjudge1 |
提出日時 | 2024-09-25 18:01:58 |
言語 | C++17(clang) (17.0.6 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,801 bytes |
コンパイル時間 | 2,590 ms |
コンパイル使用メモリ | 162,384 KB |
実行使用メモリ | 9,008 KB |
最終ジャッジ日時 | 2024-09-25 18:02:05 |
合計ジャッジ時間 | 5,886 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 36 ms
6,528 KB |
testcase_01 | AC | 35 ms
6,656 KB |
testcase_02 | AC | 37 ms
6,656 KB |
testcase_03 | AC | 36 ms
6,528 KB |
testcase_04 | AC | 37 ms
6,528 KB |
testcase_05 | AC | 37 ms
6,528 KB |
testcase_06 | AC | 37 ms
6,528 KB |
testcase_07 | AC | 37 ms
6,528 KB |
testcase_08 | AC | 58 ms
7,912 KB |
testcase_09 | AC | 58 ms
8,036 KB |
testcase_10 | AC | 58 ms
8,036 KB |
testcase_11 | AC | 59 ms
8,032 KB |
testcase_12 | AC | 58 ms
8,036 KB |
testcase_13 | AC | 61 ms
8,296 KB |
testcase_14 | AC | 70 ms
8,292 KB |
testcase_15 | AC | 60 ms
8,164 KB |
testcase_16 | AC | 63 ms
8,292 KB |
testcase_17 | AC | 61 ms
8,288 KB |
testcase_18 | AC | 113 ms
8,372 KB |
testcase_19 | AC | 112 ms
8,364 KB |
testcase_20 | WA | - |
testcase_21 | WA | - |
testcase_22 | AC | 79 ms
8,420 KB |
testcase_23 | AC | 153 ms
9,008 KB |
testcase_24 | AC | 154 ms
9,008 KB |
testcase_25 | AC | 155 ms
8,880 KB |
testcase_26 | AC | 153 ms
8,880 KB |
testcase_27 | AC | 154 ms
8,880 KB |
testcase_28 | AC | 157 ms
8,872 KB |
testcase_29 | AC | 37 ms
6,656 KB |
testcase_30 | AC | 36 ms
6,656 KB |
ソースコード
#include<bits/stdc++.h> using namespace std; const int MAXN = 200005, MOD = 998244353; int n, a[MAXN], f[MAXN], ans, Max, cnt, sum[MAXN], res[MAXN], inv[MAXN], Inv[MAXN]; vector<int> pos; int A(int n, int m) { return (n < m ? 0 : 1ll * f[n] * Inv[n - m] % MOD); } int C(int n, int m) { return (n < m ? 0 : 1ll * f[n] * Inv[m] % MOD * Inv[n - m] % MOD); } int Pow(int a, int b) { int ret = 1; for(; b; a = 1ll * a * a % MOD, b >>= 1) { if(b & 1) { ret = 1ll * ret * a % MOD; } } return ret; } int Calc(int n, int x, int a) { return 1ll * (1ll * f[a] * (x - a) % MOD * Pow(f[a - x], MOD - 2) % MOD + 1ll * f[n + 1] * (n - x + 1) % MOD * Pow(f[n - x + 1], MOD - 2) % MOD) % MOD * Pow(x + 1, MOD - 2) % MOD; } int main() { ios::sync_with_stdio(false), cin.tie(0), cout.tie(0); cin >> n; f[0] = inv[1] = Inv[0] = res[0] = 1; for(int i = 1; i < MAXN; ++i) { f[i] = 1ll * f[i - 1] * i % MOD; res[i] = (res[i - 1] + Pow(f[i], MOD - 2)) % MOD; inv[i] = (i > 1 ? 1ll * (MOD - MOD / i) * inv[MOD % i] % MOD : 1); Inv[i] = 1ll * Inv[i - 1] * inv[i] % MOD; } for(int i = 1; i <= n; ++i) { cin >> a[i]; if(a[i] != -1) { sum[a[i]] = 1; } } for(int i = 1; i <= n; ++i) { if(!sum[i]) { pos.emplace_back(i); } sum[i] = sum[i - 1] + !sum[i]; } for(int i = 1; i <= n; ++i) { Max = max(Max, a[i]), cnt += (a[i] == -1); if(a[i] == -1) { int x = lower_bound(pos.begin(), pos.end(), Max) - pos.begin(); if(x == pos.size()) { continue; } ans = (ans + 1ll * Calc(sum[n] - 1, cnt - 1, max(cnt - 1, x)) % MOD * f[sum[n] - cnt] % MOD) % MOD; }else if(a[i] == Max) { ans = (ans + 1ll * A(sum[Max], cnt) * f[sum[n] - cnt] % MOD) % MOD; } } cout << ans; return 0; }