結果

問題 No.2126 MEX Game
ユーザー HIcoder
提出日時 2023-07-11 11:41:37
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,808 bytes
コンパイル時間 683 ms
コンパイル使用メモリ 100,100 KB
最終ジャッジ日時 2025-02-15 09:43:12
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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);


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

    const int MAXN=100000;
    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]]++;
    }

    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[k]++;
        }else if(cnt[k]==1){
            cnt1[k]++;
        }else if(cnt[k]==2){
            cnt2[k]++;
        }else if(cnt[k]>2){
            cntlargerthan2[k]++;
        }
    }
    for(int i=1;i<=MAXN;i++){
        cnt0[i]=cnt0[i-1]+cnt0[i];//cnt0[i]=0~iのなかで　カウント0の個数
        cnt1[i]=cnt1[i-1]+cnt1[i];
        cnt2[i]=cnt2[i-1]+cnt2[i];
        cntlargerthan2[i]=cntlargerthan2[i-1]+cntlargerthan2[i];

    }

    int ans=0;

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

        if((cnt2[k-1]+cntlargerthan2[k-1])==k){
            //答えはk以上にできる
            ans=k;
            continue;
        }else if(cnt0[k-1]>0 || cnt1[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]);

            if(cntlargerthan2[k-1]>2 || num>=1){
                ans=k;
            }else{
                ans=k-1;
                break;
            }

        }
    }

    cout<<ans<<endl;

}
0