L2-1 包装机 (25 分)(STL43行代码)

L2-1 包装机 (25 分)

一种自动包装机的结构如图 1 所示。首先机器中有 N
条轨道，放置了一些物品。轨道下面有一个筐。当某条轨道的按钮被按下时，活塞向左推动，将轨道尽头的一件物品推落筐中。当 0号按钮被按下时，机械手将抓取筐顶部的一件物品，放到流水线上。

一种特殊情况是，因为筐的容量是有限的，当筐已经满了，但仍然有某条轨道的按钮被按下时，系统应强制启动 0号键，先从筐里抓出一件物品，再将对应轨道的物品推落。此外，如果轨道已经空了，再按对应的按钮不会发生任何事；同样的，如果筐是空的，按 0号按钮也不会发生任何事。

现给定一系列按钮操作，请你依次列出流水线上的物品。

输入格式：

输入第一行给出 3 个正整数 N（≤100）、M（≤1000）和 S​max ​​ （≤100），分别为轨道的条数（于是轨道从 1 到 N
编号）、每条轨道初始放置的物品数量、以及筐的最大容量。随后 N 行，每行给出 M 个英文大写字母，表示每条轨道的初始物品摆放。
最后一行给出一系列数字，顺序对应被按下的按钮编号，直到 −1
标志输入结束，这个数字不要处理。数字间以空格分隔。题目保证至少会取出一件物品放在流水线上。

输出格式：

在一行中顺序输出流水线上的物品，不得有任何空格。

输入样例：

3 4 4
GPLT
PATA
OMSA
3 2 3 0 1 2 0 2 2 0 -1

输出样例：

MATA

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n,m,smax;
	cin>>n>>m>>smax;
	queue<char>a[n+1];//模拟队列，n+1是因为后面更加方便
	stack<char>b;//用栈模拟筐子
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<m;j++)//读取n条轨道的m个元素
		{
			char tem;cin>>tem;
			a[i].push(tem);
		}
	
	}
	
	while(1)
	{
		int x;
		cin>>x;
		if(x==-1)//如果输入-1正常退出
			break;
		if(x==0&&!b.empty())//为零且存在元素就出栈，如果没有判空会出现段错误！！
		{
			cout<<b.top();
			b.pop();
		}
		if(x>0&&b.size()<smax&&a[x].size())//入栈
		{
			b.push(a[x].front());
			a[x].pop();
		}
		else if(x>0&&b.size()>=smax&&a[x].size())//如果栈满还继续加入元素，就先拿出一个，再加入元素
												//这里的是else if语句，不然会和上一条重复操作。					             												
		{
			cout<<b.top();
			b.pop();
			b.push(a[x].front());
			a[x].pop();
		}
	}
}