PAT (Basic Level) Practice 1072

1072 开学寄语（20 分）

下图是上海某校的新学期开学寄语：天将降大任于斯人也，必先删其微博，卸其 QQ，封其电脑，夺其手机，收其 ipad，断其 wifi，使其百无聊赖，然后，净面、理发、整衣，然后思过、读书、锻炼、明智、开悟、精进。而后必成大器也！

本题要求你写个程序帮助这所学校的老师检查所有学生的物品，以助其成大器。

输入格式：

输入第一行给出两个正整数 N（≤ 1000）和 M（≤ 6），分别是学生人数和需要被查缴的物品种类数。第二行给出 M 个需要被查缴的物品编号，其中编号为 4 位数字。随后 N 行，每行给出一位学生的姓名缩写（由 1-4 个大写英文字母组成）、个人物品数量 K（0 ≤ K ≤ 10）、以及 K 个物品的编号。

输出格式：

顺次检查每个学生携带的物品，如果有需要被查缴的物品存在，则按以下格式输出该生的信息和其需要被查缴的物品的信息（注意行末不得有多余空格）：

姓名缩写: 物品编号1 物品编号2 ……

最后一行输出存在问题的学生的总人数和被查缴物品的总数。

输入样例：

4 2
2333 6666
CYLL 3 1234 2345 3456
U 4 9966 6666 8888 6666
GG 2 2333 7777
JJ 3 0012 6666 2333

输出样例：

U: 6666 6666
GG: 2333
JJ: 6666 2333
3 5

分析：本来打算用unordered_map<string, vector<int> >直接一步到位。自己测试感觉可以，但是根本通过不了。所以这里还是用了一堆vector来存人名和物品。最后输出要记得用printf()来输出四位数。

代码： 

#include<iostream>
#include<vector>
#include<map>
#include<string>
using namespace std;
vector<vector<int> > item;
vector<string> stu;
map<int, bool> m;
int main() {
	//freopen("D:\\算法笔记\\例题\\PAT\\input.txt", "r", stdin);
	int N, M;
	cin >> N >> M;
	for (int i = 0; i < M; i++) {
		int temp;
		cin >> temp;
		m[temp] = true;
	}
	getchar();
	int sum = 0;
	for (int i = 0; i < N; i++) {
		string name;
		int num;
		bool isValid = true;
		vector<int> t;
		cin >> name >> num;
		for (int j = 0; j < num; j++) {
			int temp;
			cin >> temp;
			if (m[temp] == true) {
				isValid = false;
				t.push_back(temp);
				sum++;
			}
		}
		if (isValid == false) {
			stu.push_back(name);
			item.push_back(t);
		}
	}
	for (int i = 0; i < item.size(); i++) {
		cout << stu[i] << ": ";
		for (int j = 0; j < item[i].size(); j++) {
			if (j != item[i].size() - 1) {
				printf("%04d ", item[i][j]);
			}else {
				printf("%04d\n", item[i][j]);
			}
		}
	}
	cout << stu.size() << " " << sum << endl;
	return 0;
}