加速清洗

练一练1 加速清洗

时间限制：1s内存限制：128M

 题目描述

皮皮又在清洗餐盘！
这次为了让餐盘的清洗工作尽快结束，当待清洗的餐盘大于等于 3 时，皮皮会一次清洗两个餐盘！
开始时，没有待清洗餐盘，之后会发生两种事件：
1.新餐盘加入：我们读入一个正整数 a 时，表示编号为 a 的餐盘放在餐盘顶端；
2.清洗餐盘：我们读入一个 0 时，若待清洗餐盘数不小于 3，则先清洗排在顶端的两个餐盘，否则只清洗顶端的一个餐盘。
两种事件一共发生了 n 次，请输出每次清洗的餐盘编号吧！

【输入格式】

输入共 2 行：

第 1 行，一个正整数 n，为事件发生了次数；

第 2 行，n 个非负整数 a_1,a_2,…,a_na1​,a2​,…,an​，若 a_iai​ > 0，事件是 a_iai​ 号餐盘放在餐盘顶端；若 a_iai​ = 0，事件是清洗排在顶端的 1～2 个餐盘。

【输出格式】

输出若干行：

每行一个正整数，每当排在顶端的餐盘清洗时，在一行中输出它的编号。

【输入输出样例#1】

输入#1

复制

8
1 2 3 0 4 0 5 6

输出#1

复制

3
2
4

【输入输出样例#2】

输入#2

复制

5
3 2 1 0 0

输出#2

复制

1
2
3

【数据范围】

1 ≤ n ≤ 1000，0 ≤ a_1,a_2,…,a_na1​,a2​,…,an​ ≤ 1,000,000。

#include <iostream>
#include <stack>
using namespace std;

stack<int> st; 

int main()
{
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        int a;
        cin >> a;
        if (a ==0) 
        {  
            if (st.size()<3) 
            {  
                cout<<st.top()<<endl;
                st.pop();
            }
            else
            {
                cout<<st.top()<<endl;
                st.pop();
                cout<<st.top()<<endl;
                st.pop();
            }
        }
        else
        {
            st.push(a);
        }
    }
    
    return 0;
}
小猴编程（12950641953687092）