原题链接: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;
}
有不对的地方,欢迎指出