盲盒包装流水线

注意本题要把stack容器容量拉到最大不然最后两个测试点过不去会有段错误

众所周知，PAT 有 9 枚徽章，分别对应青铜、白银、黄金、白金、钻石、大师、王者、大圣、天神这 9 个段位，只有成绩非常优秀的考生才有资格获得刻有自己名字的徽章。现在，PAT 制作了徽章的小型纪念版，要制成盲盒给大家玩了！
下图是一条盲盒包装流水线的示意图。首先徽章通过进货口被压入货栈里，空盒在履带上从左向右传送。每次从货栈里弹出一枚徽章，进入打包机，装入一只空盒，打包后继续向右边传送。当货栈为空时，打包机会暂停，等待下一批徽章压入货栈。
每只盒子都有一个编号，小拼姐姐手里有进入流水线的空盒编号顺序表，也有每一批送往货栈的徽章顺序表，这样她其实可以知道每只盒子里装了哪种徽章。有些小朋友收到了盲盒，就想在拆封前问无所不知的小拼姐姐，盒子里的徽章是哪一种。但是因为盲盒总量可能有10^5这么多，小拼姐姐可记不住每只盒子里装的是什么，于是你就被请来写个程序帮小拼姐姐回复这种信息。
输入格式：
输入第一行给出 2 个正整数，分别为盲盒总量 N（≤10^5）和货栈容量 S（≤100）。接下来一行给出 N 只盒子的编号，编号由 5 位数字组成，给出的顺序是空盒进入传送带的顺序。随后 N/S（保证是整数）行，每行给出一批 S 枚徽章的类型，为 1-9 的数字，给出的顺序是从进货口入栈的顺序。
再下面给出一个正整数 K（≤10^4），为查询次数。随后 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<bits/stdc++.h>
using namespace std;
const int N = 1e6+3;
// typedef struct H{
//     int id;
//     int duwe;
// }He;
int h[N];
set<int> hh;
map<int,int> mp;
stack<int> st[39003];//注意这里其不卡最大值不行
int main(){
    int n,m;
    cin>>n>>m;
    for(int i = 0;i<n;i++){
        int s;
        cin>>s;
        //mp[s] = -1;
        h[i] = s;
        hh.insert(s);
        //h[i].duwe = -1;
    }
    int u = n/m;
    for(int i = 0;i<u;i++){
        for(int j = 0;j<m;j++){
            int s;
            cin>>s;
            st[i].push(s);
        }
    }
    int b = 0;
        for(int j = 0;j<u;j++){
            // for(int k = 0;k<m;k++){
            //     //cout<<h[b]<<" ";
            //     mp[h[b]]= st[j].top();
            //     st[j].pop();
            //     b++;
            // }
            while(!st[j].empty()){
                //cout<<h[b]<<" ";
                mp[h[b]]= st[j].top();
                st[j].pop();
                b++;
            }
        }
    int num;
    cin>>num;
    for(int i = 0;i<num;i++){
        int s;
        cin>>s;
        if(hh.find(s)!=hh.end()){
            cout<<mp[s]<<endl;
        }
        else{
            cout<<"Wrong Number"<<endl;
        }
    }
    return 0;
}