寝癖頭の解法

小学生の目線から、勉強中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #C - カードゲーム

第7回日本情報オリンピック 予選(過去問)から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
競技プログラミング」を略して、「競プロ」などと呼ばれています。

#C - カードゲーム

atcoder.jp

僕が作成、提出したコードは、以下のとおりです。

/*
AtCoder Problems in C++
#C - カードゲーム
https://atcoder.jp/contests/joi2008yo/tasks/joi2008yo_c
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
  int n;
  cin>>n;
  set<int> t,h;
  for(int i=1;i<=2*n;i++){
    h.insert(i);
  }
  vector<int> v(n);
  for(int i=0;i<n;i++){
    cin>>v[i];
    t.insert(v[i]);
    h.erase(v[i]);
  }
  int card=0;
  while(!(h.empty())){
    auto itr=t.lower_bound(card);
    if(t.end()==itr){
      card=0;
    }else{
      card=*itr;
      t.erase(itr);
    }
    if(t.empty()){
      break;
    }
    itr=h.lower_bound(card);
    if(h.end()==itr){
      card=0;
    }else{
      card=*itr;
      h.erase(itr);
    }
  }
  cout<<h.size()<<endl;
  cout<<t.size()<<endl;
  return 0;
}