锦绣育才【集训题】包装机

出题人：

描述

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

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

image.pngimage.png

图 2 顺序按下按钮 3、2、3、0、1、2、0 后包装机的状态
一种特殊情况是，因为筐的容量是有限的，当筐已经满了，但仍然有物品被推落时，系统会先强制启动 0 号键，从筐里抓出一件物品，再将对应轨道的物品推落。
此外，如果轨道已经空了，再按对应的按钮不会有物品被推落；
同样的，如果筐是空的，按 0 号按钮也不会发生任何事。
现给定一系列按钮操作，请你依次列出流水线上的物品。

输入描述

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

输出描述

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

用例输入 1

3 4 4
GPLT
PATA
OMSA
3 2 3 0 1 2 0 2 2 0 -1
用例输出 1

MATA

#include <iostream>
#include <stack>
#include <vector>
using namespace std;

int main() {
    int N, M, Smax;
    cin >> N >> M >> Smax;
    vector<stack<char>> tracks(N);
    for (int i = 0; i < N; ++i) {
        string items;
        cin >> items;
        for (int j = M - 1; j >= 0; --j) {
            tracks[i].push(items[j]);
        }
    }
    stack<char> basket;
    vector<char> result;
    int button;
    while (cin >> button && button!= -1) {
        if (button == 0) {
            if (!basket.empty()) {
                result.push_back(basket.top());
                basket.pop();
            }
        } else {
            if (!tracks[button - 1].empty()) {
                if (basket.size() == Smax) {
                    result.push_back(basket.top());
                    basket.pop();
                }
                basket.push(tracks[button - 1].top());
                tracks[button - 1].pop();
            }
        }
    }
    for (char item : result) {
        cout << item;
    }
    return 0;
}