PTA 7-4 包装机（单调栈）

news2024/7/6 18:49:40

题目

一种自动包装机的结构如图 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）和 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

题解（PTA的在线训练平台很多题java都超时，只能用c++）

#include <iostream>
#include <vector>
#include <string>

using namespace std;

const int N = 105;
const int M = 1005;
const int S = 105;
int n, m, s, bap;
int p[N];
string st[N][M];
string ba[S];
vector<int> list;
vector<string> ans;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    fill(p, p + N, 1);

    cin >> n >> m >> s;

    for (int i = 1; i <= n; i++) {
        string tc;
        cin >> tc;
        for (int j = 1; j <= m; j++) {
            st[i][j] = string(1, tc[j - 1]);
        }
    }

    int temp;
    cin >> temp;
    while (temp != -1) {
        list.push_back(temp);
        cin >> temp;
    }

    for (int i : list) {
        if (i == 0) {
            if (bap != 0) {
                ans.push_back(ba[bap--]);
            } else continue;
        } else {
            if (p[i] == m + 1) continue;
            else {
                if (bap == s) {
                    ans.push_back(ba[bap--]);
                    ba[++bap] = st[i][p[i]];
                    p[i]++;
                } else {
                    ba[++bap] = st[i][p[i]];
                    p[i]++;
                }
            }
        }
    }

    for (string c : ans) {
        cout << c;
    }

    return 0;
}