2021天梯赛 L2-037 包装机题解

L2-037 包装机 (25 分)

题目：

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

图1 自动包装机的结构

图 2 顺序按下按钮 3、2、3、0、1、2、0 后包装机的状态

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

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

输入格式：

输入第一行给出 3 个正整数 N（≤100）、M（≤1000）和 Smax（≤100），分别为轨道的条数（于是轨道从 1 到 N 编号）、每条轨道初始放置的物品数量、以及筐的最大容量。随后 N 行，每行给出 M 个英文大写字母，表示每条轨道的初始物品摆放。

输出格式：

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

输入样例 ：

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

输出样例 ：

MATA

思路：

• 用一个数组pos来记录每一行当前推了多少个字母，同时用一个空字符串ans来存储我们的答案，根据对输入的数字按照题意进行模拟即可。
• 当输入的数字x为0时，我们首先判断当前的筐box中是否还有字母，如果没有则空过，如果有，就令其出栈同时加入ans字符串中
• 如果输入的数字x不为0，我们就判断pos[x]是否小于等于列数m，如果没有这一步的话没法过全部测试点，若小于等于，就令g[x][pos[x]]入栈，同时将pos[x]加1，为下次做准备。
• 最后当输入为-1时令循环结束，再输出ans字符串即可。

代码：

#include<iostream>
#include<stack>
using namespace std;
int pos[1005];
char g[105][1005];
stack<char> box;
string ans="";
int main()
{
    int n,m,capacity;
    cin>>n>>m>>capacity;
    for(int i=1;i<=n;i++)
        pos[i]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            cin>>g[i][j];
        }
    }
    int x;
    while(cin>>x&&x!=-1)
    {
        if(x!=0)
        {
            if(pos[x]<=m)
            {
                if(box.size()==capacity)
                {
                     ans+=box.top();
                     box.pop();
                }
                box.push(g[x][pos[x]]);
                pos[x]++;
            }
        }
        else
        {
            if(!box.empty())
            {
                ans+=box.top();
                box.pop();
            }
        }
    }
    cout<<ans;
}