Problem:
碁石ならべ

問題

白と黒の碁石をテーブルの上にならべて遊ぶ.まずテーブルの左端に碁石を置く.次に左から 2 番目の場所に碁石を置く.これを n 回繰り返して n 個の碁石を横一列にならべる.ただし,新しく i 番目の碁石を置く際には,次のルールに従ってテーブル上の碁石を置き換える.

i が奇数の場合: テーブルに置いてあった碁石は置き換えず,新しい碁石を左
から i 番目に置く.

i が偶数の場合: 新しく左から i 番目に置く碁石の色とテーブル上の右端の碁
石の色が同じ場合は,テーブル上の碁石は置き換えず,新しい碁石を左から i
番目に置く.そうでない場合,すなわち,新しく左から i 番目に置く碁石の色
とテーブル上の右端の碁石の色が異なる場合は,まずテーブル上の右端の連続
した同色の碁石を全て取り除き,i 番目の碁石と同色の碁石に置き換える.そ
してテーブルの右端に i 番目の碁石を置く.

例えば,最初の 7 個の碁石を置いた時点で,

○○●●○○○

となっていたとする. (○は白の碁石を,●は黒の碁石を表す. )

8 番目の碁石が白(○)の場合は,右端の碁石と同色なので,そのまま置く.し
たがって,テーブル上の碁石は

○○●●○○○○

となる.

8 番目の碁石が黒(●)の場合は,右端の碁石(○)と色が異なるので,まず
テーブルの右端にある 3 個の連続した白い碁石(○)を取り除き,黒い碁石
(●)に置き換える.そして右端に 8 番目の碁石を置く.したがって,テーブ
ル上の碁石は

○○●●●●●●

となる.

入力として置く碁石の順番が与えられたとき,n 個の碁石をならべ終わった後,テー
ブル上に置いてある白い碁石の個数を求めるプログラムを作成せよ.

入力

入力は複数のデータセットからなる．各データセットは以下の形式で与えられる．

1 行目には正整数 n (1 ≤ n ≤ 100000) が書かれている.2 行目以降の第 i + 1 行目 (1 ≤ i ≤ n) には,i 番目に置く碁石の色を表す整数 ci が書かれており,ci が 0 なら i 番目に置く碁石の色が白であることを,1 なら i 番目に置く碁石の色が黒であることを表す.

採点用データのうち, 配点の 50% 分については, n ≤ 10000 を満たす.

n が 0 のとき入力の終了を示す. データセットの数は 10 を超えない．

出力

データセットごとに, n 個の碁石をならべ終わった後にテーブル上に置いてある白い碁石の個数を 1 行に出力する.

入出力例

入力例

8
1
0
1
1
0
0
0
0
8
1
0
1
1
0
0
0
1
0

出力例

6
2

上記問題文と自動審判に使われるデータは、情報オリンピック日本委員会が作成し公開している問題文と採点用テストデータです。

Solution in C++ (judge verdict: Wrong Answer)

#include <iostream>
#include <sstream>
#include <fstream>
#include <iomanip>
#include <numeric>
#include <map>
#include <vector>
#include <string>
#include <vector>
#include <cassert>
#include <cmath>
#include <stack>
#include <cctype>
#include <queue>
#include <algorithm>
#include <memory>
#include <memory.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
typedef pair<int, P> PP;
const int INF = 1 << 30;
const double EPS = 1e-10;
const double PI = 3.14159;

int main(){
	int n;
	while(cin >> n && n){
		int w_count = 0;
		int b_count = 0;
		stack<int> w, b;
		w.push(-1), b.push(-1);
		for(int i = 0; i < n; i++){
			int go;
			cin >> go;
			if(i % 2 == 0){
				if(go == 1){
					w_count++;
					if(w.empty()) w.push(i);
					else{
						int lw = w.top();
						if(lw == i-1) w.pop();
						w.push(i);
					}
				}else{
					b_count++;
					if(b.empty()) b.push(i);
					else{
						int lb = b.top();
						if(lb == i-1) b.pop();
						b.push(i);
					}
				}
			}else{
				if(go == 0){
					if(b.empty()){
						b.push(i);
						b_count++;
					}else{
						int lb = b.top();
						if(lb == i-1){
							b_count++;
							b.pop();
							b.push(i);
						}else{
							w.pop();
							b.pop();
							b_count += i - lb;
							w_count -= i - lb - 1;
							b.push(i);
						}
					}
				}else{
					if(w.empty()){
						w.push(i);
						w_count++;
					}else{
						int lw = w.top();
						if(lw == i-1){
							w_count++;
							w.pop();
							w.push(i);
						}else{
							w.pop();
							b.pop();
							w_count += i - lw;
							b_count -= i - lw - 1;
							w.push(i);
						}
					}
				}
			}
		}
		cout << b_count << endl;
	}
	return 0;
}