L2-037 包装机 (25 分) （结构体+模拟栈）

题目详情 - L2-037 包装机 (25 分) (pintia.cn)

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

图1 自动包装机的结构

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

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

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

输入格式：

输入第一行给出 3 个正整数 N（≤100）、M（≤1000）和 Smax​（≤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;
struct line
{
    string l;    //l为这条轨道的物品
    int cnt;     //cnt为这条轨道的物品个数
    int now;     //now为当前准备取的物品位置，这里要注意now==cnt时代表轨道已经空了，因为now是从0开始
};
struct line t[1000];
int main ()
{
    int n,m,s,num = 0;    //num为栈中元素个数
    cin >> n >> m >> s;
    string ans;   //流水线
    char p[s+10];    //定义栈p
    for (int i = 1 ; i <= n ; i ++)
    {
        cin >> t[i].l;
        t[i].cnt = t[i].l.size();
    }
    for (int i = 0 ; ; i ++)
    {
        int k;
        scanf("%d",&k);
        if (k == -1) break;
        if (k == 0 && num > 0) ans += p[num--];    //如果为0，且p栈中有元素
        else if (k > 0)
        {
            if (t[k].now < t[k].cnt && num < s) p[++num] = t[k].l[t[k].now++];    //如果当前轨道的下标小于轨道长度，且栈未满，则往栈中加入当前轨道准备取的物品
            else if (t[k].now < t[k].cnt && num == s)    //如果当前轨道的下标小于轨道长度，且栈满，则把栈顶元素加入ans流水线中，并把栈顶元素替换成当前轨道准备取的物品
            {
                ans += p[num];
                p[num] = t[k].l[t[k].now++];
            }
        }
    }
    cout << ans;
}