PTA L2-037 包装机

题目重现：

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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

栈限制

8192 KB

实现思路：

1. 队列初始化与填充：

   • 定义了一个二维队列数组 k，大小为 [200]，用于存储 n 个队列。
   • 通过两层循环，读取每个队列的 m 个字符，并依次压入对应的队列中。

2. 操作循环：

   • 进入一个无限循环，读取一个整数 o，根据 o 的值执行不同的操作：
     • 如果 o == -1，则跳出循环，结束程序。
     • 如果 o == 0，从栈顶弹出一个元素并打印，但确保栈非空。
     • 否则，o 表示一个队列的编号，从该队列中取出一个元素并尝试压入栈中：
       • 如果栈已满（s.size() >= sm），则同样从栈顶弹出并打印一个元素，再将队列元素压入栈。
       • 如果队列非空，将队列的队首元素移至栈顶。

3. 数据结构：

   • 使用 stack<char> 来实现栈，queue<char>[] 来实现多个队列。
   • 栈用于存储和管理从队列中选取的元素，队列则用于存储预先给定的字符序列。

4. 输出：

   • 打印的时机有两种：当从栈中弹出元素时，或是栈满时为了腾出空间而弹出元素。

关键代码：

while(1){
        int o;
        cin>>o;
        if(o==-1)break;
        if(o==0){
            if(s.size()){
                cout<<s.top();
                s.pop();
            }
        }
        else{
            if(k[o].size()){
                if(s.size()>=sm)
                    {
                        cout<<s.top();
                        s.pop();
                    }
                    s.push(k[o].front());
                    k[o].pop();
            }
        }
    }

代码关键：

1. 操作循环:

   • 程序进入一个无限循环，读取一个整数 o 来指示下一步的操作。
   • 根据 o 的值执行不同的操作：
     • -1: 结束循环，从而终止程序。
     • 0: 当栈不为空时，从栈顶弹出一个元素并打印。
     • 非0: 尝试从编号为 o 的队列中取出一个元素并压入栈中，如果栈已满，先从栈顶弹出并打印一个元素，再压入新元素。

2. 条件判断和边界检查:

   • 每次尝试向栈中压入元素前，都会检查栈是否已满 (s.size() >= sm)，确保不会超出栈的容量限制。
   • 在从栈中弹出元素之前，也会检查栈是否为空，避免发生未定义行为。

3. 数据流和控制流:

   • 数据流包括从队列到栈的元素流动，以及从栈到控制台的打印输出。
   • 控制流由 o 的值决定，决定了程序执行哪一部分逻辑。

通过截图：

完整代码：

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n,m,sm;
    cin>>n>>m>>sm;
    stack<char> s;
    queue<char> k[200];
    for(int i=1;i<=n;i++)
    {
        char q;
        for(int j=0;j<m;j++)
        {
            cin>>q;
            k[i].push(q);
        }
    }
    while(1){
        int o;
        cin>>o;
        if(o==-1)break;
        if(o==0){
            if(s.size()){
                cout<<s.top();
                s.pop();
            }
        }
        else{
            if(k[o].size()){
                if(s.size()>=sm)
                    {
                        cout<<s.top();
                        s.pop();
                    }
                    s.push(k[o].front());
                    k[o].pop();
            }
        }
    }
    return 0;
}