PTA(栈和队列)2-7 包装机

2-7 包装机

分数 6

作者 陈越

单位 浙江大学

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

图1.JPG

图1 自动包装机的结构

图2.JPG

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

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

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

输入格式:

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

最后一行给出一系列数字,顺序对应被按下的按钮编号,直到 −1 标志输入结束,这个数字不要处理。数字间以空格分隔。题目保证至少会取出一件物品放在流水线上。

输出格式:

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

输入样例:

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

输出样例:

1
MATA

分析一下:

  • 轨道上面的先输入的先输出,所以用队列实现
  • 筐是先进后出,用栈实现
  • 机械手是抓取某个队列的头部,相当于getTop,然后pop

我的部分正确

测试点 提示 内存(KB) 用时(ms) 结果 得分
0 488 2 答案正确 1 / 1
1 556 2 答案正确 1 / 1
2 576 2 答案错误 0 / 1
3 516 2 答案错误 0 / 1
4 580 2 答案正确 1 / 1
5 972 142 段错误 0 / 1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include <iostream>
#include <stack>
#include <queue>
#include <vector>

using namespace std;

int main() {
stack<char> stack; // 模拟框子
int N, M, S;
cin >> N >> M >> S;
vector<queue<char>> vector; // 用容器存放队列
while (N--) { // 输入N条流水线上的物品
string str; // 读取M个大写字母
cin >> str;
queue<char> queue;
for (char c : str) {
queue.push(c);
}
vector.push_back(queue); // 把该队列放入容器
}
int num;
while (cin) {
cin >> num;
if (num == -1) {
break;
}
if (num != 0) {
if (stack.size() == S) { //当筐已经满了,但仍然有某条轨道的按钮被按下时,系统应强制启动 0 号键,先从筐里抓出一件物品,再将对应轨道的物品推落
cout << stack.top();
stack.pop();
}
stack.push(vector[num-1].front()); // 将指定队列的队头获取,放入栈中
vector[num-1].pop(); // 出队
} else { // 如果num为0,将从栈顶出栈
if (stack.empty()) { // 如果栈为空
continue;
} else {
cout << stack.top(); // 输出栈顶
stack.pop(); // 出栈
}
}
}
}
  • 有段错误,我发现如果队列为空,则pop不了,所以我将num!=0的情况考虑全面了一点:
1
2
3
4
5
6
7
8
9
10
11
12
13
if (num != 0) {
if (stack.size() == S) { //当筐已经满了,但仍然有某条轨道的按钮被按下时,系统应强制启动 0 号键,先从筐里抓出一件物品,再将对应轨道的物品推落
cout << stack.top();
stack.pop();
}
// 如果队列为空,直接下一次循环
if (vector[num-1].empty()) {
continue;
} else {
stack.push(vector[num-1].front()); // 将指定队列的队头获取,放入栈中
vector[num-1].pop(); // 出队
}
}
  • 发现答案还是有一个不对:
测试点 提示 内存(KB) 用时(ms) 结果 得分
0 356 2 答案正确 1 / 1
1 304 2 答案正确 1 / 1
2 524 2 答案正确 1 / 1
3 548 2 答案正确 1 / 1
4 344 2 答案正确 1 / 1
5 704 141 答案错误 0 / 1

正确答案

  • 发现是代码顺序问题:
  • 得先判断队列是不是空,是空就下一次循环,不然先判断栈满都没用
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#include <iostream>
#include <stack>
#include <queue>
#include <vector>

using namespace std;

int main() {
stack<char> stack; // 模拟框子
int N, M, S;
cin >> N >> M >> S;
vector<queue<char>> vector; // 用容器存放队列
while (N--) { // 输入N条流水线上的物品
string str; // 读取M个大写字母
cin >> str;
queue<char> queue;
for (char c : str) {
queue.push(c);
}
vector.push_back(queue); // 把该队列放入容器
}
int num;
while (cin) {
cin >> num;
if (num == -1) {
break;
}
if (num != 0) {
// 如果队列为空,直接下一次循环
if (vector[num-1].empty()) {
continue;
}
if (stack.size() == S) { //当筐已经满了,但仍然有某条轨道的按钮被按下时,系统应强制启动 0 号键,先从筐里抓出一件物品,再将对应轨道的物品推落
cout << stack.top();
stack.pop();
}

stack.push(vector[num-1].front()); // 将指定队列的队头获取,放入栈中
vector[num-1].pop(); // 出队

} else { // 如果num为0,将从栈顶出栈
if (stack.empty()) { // 如果栈为空
continue;
} else {
cout << stack.top(); // 输出栈顶
stack.pop(); // 出栈
}
}
}
}