题目要求:
众所周知,PAT 有 9 枚徽章,分别对应青铜、白银、黄金、白金、钻石、大师、王者、大圣、天神这 9 个段位,只有成绩非常优秀的考生才有资格获得刻有自己名字的徽章。现在,PAT 制作了徽章的小型纪念版,要制成盲盒给大家玩了!
下图是一条盲盒包装流水线的示意图。首先徽章通过进货口被压入货栈里,空盒在履带上从左向右传送。每次从货栈里弹出一枚徽章,进入打包机,装入一只空盒,打包后继续向右边传送。当货栈为空时,打包机会暂停,等待下一批徽章压入货栈。
每只盒子都有一个编号,小拼姐姐手里有进入流水线的空盒编号顺序表,也有每一批送往货栈的徽章顺序表,这样她其实可以知道每只盒子里装了哪种徽章。有些小朋友收到了盲盒,就想在拆封前问无所不知的小拼姐姐,盒子里的徽章是哪一种。但是因为盲盒总量有 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
代码解析:
#include<iostream>
#include<stack>
#include<algorithm>
using namespace std;
stack<int> st;
int main()
{
int n, s, no;
int a[10000], b[10000];
cin >> n >> s;
int num = n / s;
int j = 1, ci = 1;
while (num--)
{
for (int i = 1; i <= s; i++)
{
cin >> no;
st.push(no);
}
for (; j <= ci * s; j++)
{
a[j] = st.top();
st.pop();
}
ci++;
}
j = 1;
for (; j <= n; j++)
{
cin >> num;
b[j] = num;
}
cin >> num; int* index;
for (int i = 1; i <= num; i++)
{
cin >> no;
index = find(a + 1, a + n + 1, no);
if ((index - a) > n || (index - a) < 0)
{
cout << "Wrong Number" << endl;
continue;
}
cout << b[index - a] << endl;
}
return 0;
}