結果
問題 |
No.3075 Mex Recurrence Formula
|
ユーザー |
![]() |
提出日時 | 2025-04-04 11:58:35 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 98 ms / 2,000 ms |
コード長 | 1,324 bytes |
コンパイル時間 | 3,081 ms |
コンパイル使用メモリ | 145,416 KB |
実行使用メモリ | 14,976 KB |
最終ジャッジ日時 | 2025-04-04 11:58:43 |
合計ジャッジ時間 | 7,201 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 46 |
ソースコード
#include <iostream> #include <vector> #include <algorithm> #include <set> #include <atcoder/all> using namespace std; using namespace atcoder; using ll = long long; using mint = modint998244353; using vi = vector<int>; using vvi = vector<vi>; using vvvi = vector<vvi>; using vll = vector<ll>; using vvll = vector<vll>; using vvvll = vector<vvll>; using vmi = vector<mint>; using vvmi = vector<vmi>; using vvvmi = vector<vvmi>; #define all(a) (a).begin(), (a).end() #define rep2(i, m, n) for (int i = (m); i < (n); ++i) #define rep(i, n) rep2(i, 0, n) #define drep2(i, m, n) for (int i = (m)-1; i >= (n); --i) #define drep(i, n) drep2(i, n, 0) int main(){ ios::sync_with_stdio(false); std::cin.tie(nullptr); int n; ll x; cin >> n >> x; vi a(n), cnt(n+1, 0); rep(i, n){ cin >> a[i]; int m = (a[i] < n ? a[i] : n); cnt[m]++; } if(x <= n){ cout << a[x-1] << endl; return 0; } set<int> st; rep(i, n+1){ if(cnt[i] == 0)st.insert(i); } vi b(n+1); rep(i, n+1){ b[i] = *st.begin(); cnt[b[i]]++; st.erase(b[i]); if(i == n)break; int m = (a[i] < n ? a[i] : n); cnt[m]--; if(cnt[m] == 0)st.insert(m); } ll base = x - n; cout << b[(base - 1) % (n+1)] << endl; return 0; }