Strange Balls

原题链接：https://atcoder.jp/contests/abc240/tasks/abc240_d?lang=en

第一行给一个数N

第二行有N个球，球上有一个不小于2的数字

按照给的顺序放入一个盒子中，如果一个数相连次，那么这数都要被消除。

每放入一个数，就输出盒里有多少球

 

 

//zheng.ys
#include<bits/stdc++.h>
#include<algorithm>
#define int long long
#define endl '\n'
using namespace std;

typedef pair<int,int> PII;

const int N=200010;

void solve()
{
	int n;
	int a[N];
	stack<PII>zheng;//第一个表示a[i]的值,第二个表示与a[i]相连的有几个  
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
	{
		//为空或者与上一个值不同 直接进栈  
		if(zheng.empty()||zheng.top().first!=a[i])
			zheng.push({a[i],1});
		else//zheng.top().first==a[i] 
		{
			//如果加上1的值等于a[i]，那么可以消除  
			if(zheng.top().second+1==a[i])
			{
				while(!zheng.empty()&&zheng.top().first==a[i])
					zheng.pop();
			}
			else//与a[i]相连的不够a[i]所带的值 
				zheng.push({a[i],zheng.top().second+1});
		}
		cout<<zheng.size()<<endl;
	}
}

signed main()
{
	ios::sync_with_stdio(false);
	int t=1; // cin>>t;
	while(t--)
		solve();
	
	
	return 0;
}

有不对的地方，欢迎指出