7-13 盲盒包装流水线

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

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

输出格式：

对每个查询编号，在一行中输出该盒子中装的徽章类型。如果编号是错误的，则在一行中输出 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

---

思路：

使用queue保存盒子编号；

使用stack保存徽章编号；

每输入s个徽章编号，就按要求处理，将结果保存到map中。

代码：

#include<iostream>
#include<stack>
#include<queue>
#include<map>
using namespace std;

int main()
{
	int n, s;
	cin >> n >> s;

	queue<int> q;
	int temp;
	for (int i = 0; i < n; i++) {
		cin >> temp;

		q.push(temp);
	}

	map<int, int> mp;
	stack<int> st;
	int a, b;
	for (int i = 0; i < n / s; i++) {
		for (int j = 0; j < s; j++) {
			cin >> temp;

			st.push(temp);
		}

		for (int j = 0; j < s; j++) {
			a = q.front();
			q.pop();
			b = st.top();
			st.pop();

			mp[a] = b;
		}
	}

	int k;
	cin >> k;

	for (int i = 0; i < k; i++) {
		cin >> temp;

		if (mp.find(temp) != mp.end()) {
			cout << mp[temp] << endl;
		}
		else {
			cout << "Wrong Number" << endl;
		}
	}

	return 0;
}