每日一练——盲盒包装流水线 (25 分)

523 阅读3分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第16天,点击查看活动详情。​​

题目详情:

众所周知,PAT 有 9 枚徽章,分别对应青铜、白银、黄金、白金、钻石、大师、王者、大圣、天神这 9 个段位,只有成绩非常优秀的考生才有资格获得刻有自己名字的徽章。现在,PAT 制作了徽章的小型纪念版,要制成盲盒给大家玩了!

下图是一条盲盒包装流水线的示意图。首先徽章通过进货口被压入货栈里,空盒在履带上从左向右传送。每次从货栈里弹出一枚徽章,进入打包机,装入一只空盒,打包后继续向右边传送。当货栈为空时,打包机会暂停,等待下一批徽章压入货栈。

lsx.png

每只盒子都有一个编号,小拼姐姐手里有进入流水线的空盒编号顺序表,也有每一批送往货栈的徽章顺序表,这样她其实可以知道每只盒子里装了哪种徽章。有些小朋友收到了盲盒,就想在拆封前问无所不知的小拼姐姐,盒子里的徽章是哪一种。但是因为盲盒总量有 105 这么多,小拼姐姐可记不住每只盒子里装的是什么,于是你就被请来写个程序帮小拼姐姐回复这种信息。

输入格式:

输入第一行给出 2 个正整数,分别为盲盒总量 N(≤105)和货栈容量 S(≤100)。接下来一行给出 N 只盒子的编号,编号由 5 位数字组成,给出的顺序是空盒进入传送带的顺序。随后 N/S(保证是整数)行,每行给出一批 S 枚徽章的类型,为 1-9 的数字,给出的顺序是从进货口入栈的顺序。

再下面给出一个正整数 K(≤104),为查询次数。随后 K 行,每行给出一个 5 位编号。

输出格式:

对每个查询编号,在一行中输出该盒子中装的徽章类型。如果编号是错误的,则在一行中输出 Wrong Number

输入样例:

10 5
00132 10093 92001 23333 66666 88888 09009 34658 82750 69251
1 2 3 4 5
9 8 7 6 1
5
66666
88888
69251
55555
10093

输出样例:

1
1
9
Wrong Number
4

思路:

模拟,哈希

主要考察队列和栈,直接用数组模拟也是可以的。

注意题目并没有说编号没有前导零,保险起见不用int存编号

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <unordered_map>

using namespace std;

int n, s;
vector<string> v;
vector<int> stk;
unordered_map<string, int> st;

int main() 
{
	cin >> n >> s;
	v.resize(n);
	
	for(int i = 0; i < n; i++) {
		cin >> v[i];
	}
	
	for(int i = 0, u = 0; i < n / s; i++) {
		stk.clear();
		stk.resize(s);
		for(int j = 0; j < s; j++) {
			cin >> stk[j];
		}
		
		reverse(stk.begin(), stk.end());
		for(int id : stk) {
			st[v[u++]] = id;
		}
	}
	
	int k;
	cin >> k;
	while(k--)
	{
		string id;
		cin >> id;
		if(st.count(id)) {
			cout << st[id] << endl;
		} else {
			puts("Wrong Number");
		}
	}
	return 0;
}