寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #A - 碁石ならべ

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

#A - 碁石ならべ

https://www.ioi-jp.org/joi/2007/2008-ho-prob_and_sol/2008-ho.pdf#page=2

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

/*
AtCoder Problems in C++
#A - 碁石ならべ
https://www.ioi-jp.org/joi/2007/2008-ho-prob_and_sol/2008-ho.pdf#page=2
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
  int n;
  cin>>n;
  vector<int> v;
  int num=-1;
  for(int i=1;i<=n;i++){
    int c;
    cin>>c;
    if(i%2==0){
      if(num==c){
        v.back()++;
      }else{
        int j=v.back();
        v.pop_back();
        if(v.empty()){
          v.push_back(j+1);
        }else{
          v.back()+=(j+1);
        }
      }
    }else{
      if(num==c){
        v.back()++;
      }else{
        v.push_back(1);
      }
    }
    num=c;
  }
  int ans=0;
  for(int i=v.size()-1-num;i>=0;i-=2){
    ans+=v[i];
  }
  cout<<ans<<endl;
  return 0;
}