时间限制:1.000S 空间限制:128MB
题目描述
在餐厅里,洗盘子的工作需要使用到栈这种数据结构。
假设你手里有一个盘子堆放区。现在需要模拟洗盘子的过程,每个盘子都有一个编号。
盘子堆放区操作说明:
1. 当操作为 1 时,表示从盘子堆放区拿走顶部的盘子清洗。
2. 当操作为 2 时,表示有未洗的盘子放入盘子堆放区。
在一系列操作之后,你需要回答:下一个清洗的盘子编号?
输入描述
第一行有一个整数 n,代表初始盘子堆放区中盘子的数量为 n。
第二行有 n 个整数,代表了盘子的编号,同时整数之间的顺序也代表了未洗盘子加入盘子堆放区的顺序。
第三行为一个整数 m,代表接下来将会有 m 次操作。
接下来一共有 m 行,代表共有 m 次操作。
如果是操作 1,那么该行只会有一个数字 1,代表有一个盘子被拿走清洗。
如果是操作 2,那么该行有两个数字,第一个数字 2 表示有未洗的盘子加入,第二个数字代表未洗的盘子编号。
输出描述
输出共一行,为下一个该清洗的盘子编号。 如果没有下一个该清洗的盘子,那么请输出 “All the dishes have been washed.”
输入示例
5
1001 1002 1003 1004 1005
3
1
1
2 1006
输出示例
1006
代码示例1
#include <iostream>
#include <stack>
using namespace std;
int main()
{
int n,val,opt,num,m;
cin >> n;//输入盘子的数量
stack<int> st;//初始化一个栈结构
while(n--)
{
cin >> val;//输入盘子的编号
st.push(val);//将输入的n个盘子的编号push到栈中
cin >> m;//接受m,表示m次操作
while(m--)
{
cin >> opt;接收操作的指令,1表示出栈,2表示入栈
if(opt == 1)
{
if(!st.empty())
st.pop();
}
else if(opt == 2)
{
cin >> num;
st.push(num);
}
}
if(st.empty())
{
cout <<"All the dishes have been washed."<<endl;
}
else
{
cout << st.top()<< endl;
}
}
return 0;
}
此版本虽然结果正确,但内存超限
代码示例2
#include <iostream>
#include <stack>
using namespace std;
int main()
{
int n,val,opt,num,m;
cin >> n;//输入盘子的数量
stack<int> st;//初始化一个栈结构
while(n--)
{
cin >> val;//输入盘子的编号
st.push(val);//将输入的n个盘子的编号push到栈中
}
cin >> m;//接受m,表示m次操作
while(m--)
{
cin >> opt;接收操作的指令,1表示出栈,2表示入栈
if(opt == 1 && !st.empty())
{
st.pop();
}
else if(opt == 2)
{
cin >> num;
st.push(num);
}
}
if(st.empty())
{
cout <<"All the dishes have been washed."<<endl;
}
else
{
cout << st.top()<< endl;
}
return 0;
}
此版本符合要求,经检查发现版本1中while(n--)的循环应该在st.push(val)后结束。