結果

問題 No.3780 Erase and Mex
コンテスト
ユーザー Rubikun
提出日時 2026-10-30 21:45:10
言語 C++17
(gcc 15.3.0 + boost 1.92.0 + ACL)
コンパイル:
g++-15 -O2 -fplugin=/usr/lib/yukicoder/noavx512-gcc15.so -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 2,033 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,507 ms
コンパイル使用メモリ 224,396 KB
実行使用メモリ 9,924 KB
最終ジャッジ日時 2026-10-30 21:45:16
合計ジャッジ時間 6,289 ms
ジャッジサーバーID
（参考情報） 		judge4_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 31 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return true; } return false; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return true; } return false; }
#define vi vector<int>
#define vl vector<ll>
#define vii vector<pair<int,int>>
#define vll vector<pair<ll,ll>>
#define vvi vector<vector<int>>
#define vvl vector<vector<ll>>
#define vvii vector<vector<pair<int,int>>>
#define vvll vector<vector<pair<ll,ll>>>
#define vst vector<string>
#define pii pair<int,int>
#define pll pair<ll,ll>
#define pb push_back
#define all(x) (x).begin(),(x).end()
#define mkunique(x) sort(all(x));(x).erase(unique(all(x)),(x).end())
#define fi first
#define se second
#define mp make_pair
#define si(x) int(x.size())
const int mod=998244353,MAX=300005,INF=15<<26;

ll solve(vl A){
    sort(all(A));
    ll N=si(A);
    if(N==1) return A[0];
    
    ll L=0,R=N+2;
    while(R-L>1){
        ll M=(L+R)/2;
        vl imo(M);
        imo[M-1]++;
        
        vl CN(M);
        ll rem=0;
        for(ll x:A){
            if(x>=M) rem++;
            else CN[x]++;
        }
        
        bool ok=true;
        for(ll i=M-1;i>=0;i--){
            if(i<=1) CN[i]+=rem;
            if(imo[i]>1e9){
                ok=false;
                break;
            }
            ll can=min(imo[i],CN[i]);
            ll need=imo[i]-can;
            if(i==0&&need) ok=false;
            else{
                if(i==1) rem=0;
                rem+=CN[i]-can;
                if(i) imo[i-1]+=need;
            }
            
            if(i) imo[i-1]+=imo[i];
        }
        
        if(ok) L=M;
        else R=M;
    }
    
    return L;
}
int main(){
    
    std::ifstream in("text.txt");
    std::cin.rdbuf(in.rdbuf());
    cin.tie(0);
    ios::sync_with_stdio(false);
    
    int Q;cin>>Q;
    while(Q--){
        ll N;cin>>N;
        vl A(N);
        for(int i=0;i<N;i++){
            cin>>A[i];
        }
        cout<<solve(A)<<"\n";
    }
}
0