結果

問題 No.3075 Mex Recurrence Formula
コンテスト
ユーザー Zhiyuan Chen
提出日時 2025-03-28 21:47:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 102 ms / 2,000 ms
コード長 973 bytes
コンパイル時間 2,158 ms
コンパイル使用メモリ 202,976 KB
実行使用メモリ 16,620 KB
最終ジャッジ日時 2025-03-28 21:47:10
合計ジャッジ時間 6,789 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N; long long X;
    cin >> N >> X;
    int total = 2 * N + 1;
    vector<long long> A(total);
    for(int i=0;i<N;i++){
        cin >> A[i];
    }
    vector<int> freq(N+1,0);
    for(int i=0;i<N;i++){
        if(A[i] <= N) freq[A[i]]++;
    }
    set<int> s;
    for(int x=0;x<=N;x++){
        if(freq[x]==0) s.insert(x);
    }
    for(int i = N; i < total; i++){
        int mex = *s.begin();
        A[i] = mex;
        int rem = A[i-N];
        if(rem <= N){
            freq[rem]--;
            if(freq[rem]==0) s.insert(rem);
        }
        if(mex <= N){
            if(s.find(mex)!=s.end()) s.erase(mex);
            freq[mex]++;
        }
    }
    if(X <= total) cout << A[X-1] << "\n";
    else{
        long long period = N + 1;
        long long offset = (X - N - 1) % period;
        cout << A[N + offset] << "\n";
    }
    return 0;
}
0