2021-天梯赛 L2-037 包装机（STL）

题目描述

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

解题思路

这里用到了两个STL

1. stack
   stack是一种先进后出的数据结构。stack容器允许新增元素，移除元素，取得栈顶元素。

• 将元素推入栈的操作称为: push
• 将元素推出stack的操作称为：pop
• 返回栈顶元素的操作为：top
• 判断该栈是否为空为：empty
• 返回该栈的大小为：size

2. queue

   queue也同样符合先进后出，但是queue只能由队尾进由队 首出，换言之只有在队首的元素才能被用。

• 往队尾添加元素的操作称为：push
• 将元素推出队列的操作称为：pop
• 返回队首元素的操作称为：front
• 返回队尾的操作称为：back
• 判断该队列是否为空为：empty
• 返回该队列的大小为：size

#include<bits/stdc++.h>
using namespace std;

int main()
{
	int n,m,sm;
	cin>>n>>m>>sm;
	queue<char>a[n+1];
	stack<char>b;
	for(int i=1; i<=n; i++){
		for(int j=0; j<m; j++){
			char t;
			cin>>t;
			a[i].push(t);
		}
	}
	
	for(; ;){
		int x;
		cin>>x;
		if(x==-1)
			break;
		//这里要注意判断是否为空
		if(x==0&&!b.empty()){
			cout<<b.top();
			b.pop(); 
		}
		if(x>0&&b.size()<sm&&a[x].size()){
			b.push(a[x].front());
			a[x].pop();
		}
		else if(x>0&&b.size()>=sm&&a[x].size()){
			cout<<b.top();
			b.pop();
			b.push(a[x].front());
			a[x].pop();
		}
	}
	return 0;
}