結果

問題 No.3075 Mex Recurrence Formula
ユーザー Hydrogen332
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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';
}
0