結果

問題 No.2126 MEX Game
ユーザー HIcoder
提出日時 2023-07-11 11:59:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,831 bytes
コンパイル時間 903 ms
コンパイル使用メモリ 101,608 KB
最終ジャッジ日時 2025-02-15 09:43:25
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<set>
#include<algorithm>
#include<vector>
#include<string>
#include<set>
#include<map>
#include<numeric>
#include<queue>
#include<cmath>
using namespace std;
typedef long long ll;
const ll INF=1LL<<60;
typedef pair<int,int> P;
typedef pair<int,P> PP;
const ll MOD=998244353;
const double PI=acos(-1);


template<class Type> struct binary_indexed_tree{
    int N;
    vector<Type> bit;
    
  

    binary_indexed_tree(int n):N(n+1){
        bit = vector<Type>(n+1,0);
        N=n;
    }



    void add(int x,Type a){
        x++;//1から始めるための補正
        //for(int i=x; i<=N; i+=(i&-i)) bit[i] = addition(bit[i],a);
        while(x<=N){
            //bit[x] = addition(bit[x],a);
            bit[x] =bit[x]+a;
            x += x&-x;
        }
    }

    
    Type sum(int x){
        x++;//1から始まることに注意
        Type ret=0;
        //for(int i=x; i>0; i-=(i&-i)) ret = addition(ret,bit[i]);
        while(x>0){
            ret = ret+bit[x];
            x -= x&-x;
        }
        
        return ret;
    }

    //[l,r]の範囲
    // rはN-1以下
    Type get(int l,int r){
        if(r>N) return 0;//配列の外へのアクセス
        if(l>r) return 0;//本来は l<=r となるのでおかしい
        if(l==0) return sum(r);//[0,r]//ここでoutなわけか
        else return (sum(r) - sum(l-1));
    }
};



int main(){
    int N;
    cin>>N;

    const int MAXN=100000+1;
    vector<int> cnt(100000+10,0);
    vector<int> A(N);
    for(int i=0;i<N;i++){
        cin>>A[i];
    }
    for(int i=0;i<N;i++){
        cnt[A[i]]++;
    }

    binary_indexed_tree<int> cnt0(MAXN+1),cnt1(MAXN+1),cnt2(MAXN+1),cntlargerthan2(MAXN+1);

    //vector<int> cnt0(MAXN+1,0),cnt1(MAXN+1,0),cnt2(MAXN+1,0),cntlargerthan2(MAXN+1,0);

    for(int k=0;k<=MAXN;k++){
        if(cnt[k]==0){
            cnt0.add(k,1);
        }else if(cnt[k]==1){
            cnt1.add(k,1);
        }else if(cnt[k]==2){
            cnt2.add(k,1);
        }else if(cnt[k]>2){
            cntlargerthan2.add(k,1);
        }
    }
   

    int ans=0;

   
    for(int k=1;k<=MAXN;k++){
        

        if((cnt2.get(0,k-1)+cntlargerthan2.get(0,k-1))==k){
            //答えはk以上にできる
            ans=k;
            
        }else if(cnt0.get(0,k-1)>0 || cnt1.get(0,k-1)>1){
            //答えはk未満
            ans=k-1;
            break;
        }else{
            //[k,MAXN]
            //int num=(cnt1[MAXN]+cnt2[MAXN]+cntlargerthan2[MAXN])-(cnt1[k-1]+cnt2[k-1]+cntlargerthan2[k-1]);

            int num=cnt1.get(k,MAXN)+cnt2.get(k,MAXN)+cntlargerthan2.get(k,MAXN);
            if(cntlargerthan2.get(0,k-1)>2 || num>=1){
                ans=k;

            }else{
                ans=k-1;
                break;
            }

        }
    }

    cout<<ans<<endl;

}
0