結果
問題 |
No.3075 Mex Recurrence Formula
|
ユーザー |
|
提出日時 | 2025-03-28 22:12:00 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 162 ms / 2,000 ms |
コード長 | 1,169 bytes |
コンパイル時間 | 3,461 ms |
コンパイル使用メモリ | 279,292 KB |
実行使用メモリ | 8,844 KB |
最終ジャッジ日時 | 2025-03-28 22:12:20 |
合計ジャッジ時間 | 8,822 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 46 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; #define rep(i, s, e) for (int i = (int)(s); i < (int)(e); ++i) #define all(a) (a).begin(),(a).end() #include <atcoder/fenwicktree> using namespace atcoder; int main() { cin.tie(nullptr); ll N, X; cin >> N >> X; vector<int> A(N); rep(i, 0, N) cin >> A[i]; --X; if (X < N) { cout << A[X] << '\n'; return 0; } X -= N; vector<int> cnt(N + 10, 0); rep(i, 0, N) if (A[i] < N + 10) ++cnt[A[i]]; fenwick_tree<int> BIT(N + 10); rep(i, 0, N + 10) if (cnt[i] > 0) BIT.add(i, 1); rep(i, 0, N + 1) { int ng = -1, ok = N + 5; while (ok - ng > 1) { int mid = (ok + ng) / 2; if (BIT.sum(0, mid + 1) == mid + 1) ng = mid; else ok = mid; } A.push_back(ok); if (ok < N + 10) { ++cnt[ok]; if (cnt[ok] == 1) BIT.add(ok, 1); } if (A[i] < N + 10) { --cnt[A[i]]; if (cnt[A[i]] == 0) BIT.add(A[i], -1); } } cout << A[X % (N + 1) + N] << '\n'; }