PAT乙 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

解题思路:

这题可以把违规物品的ID作为flag数组的下标，如果该物品违规，则对应flag数组的下标置为1，这样做的好处就在于后面每次判断该ID是否违法，可以通过flag[ID]是否为1来判断，这样操作的时间复杂度为O(1)，大大加快了效率。

程序：

#include <iostream>
#include <string>
#include <stdio.h>
using namespace std;

int main(int argc, char const *argv[])
{
	int N, M, K, thingID, cnt = 0, cntStu = 0;
	cin >> N >> M;
	int ID,flag[10001] = {0}; // flag初始化为0,如果为1说明是违规
	string name[N];
	for (int i = 0; i < M; i++){
		cin >> ID;
		flag[ID] = 1;		// 以ID作为下标
	}
	for (int i = 0; i < N; i++)
	{
		int flag2 = 0;		// 记录学生是否违规
		cin >> name[i] >> K;
		for (int j = 0; j < K; j++)
		{
			cin >> thingID;
			if (flag[thingID] != 0)
			{
				if (flag2 == 0)	// 如果是第一进入if条件,则输出姓名缩写
				{		
					cout << name[i] << ":";
					cntStu++;
				}
				flag2 = 1;
				cnt++;
				printf(" %04d", thingID);	// 按要求输出，保持4位
			}
		}
		if (flag2)		// 如果该学生违规，继续打一个回车，保持格式
			cout << endl;
	}
	cout << cntStu << " " << cnt << endl;
	return 0;
}

如果对您有帮助，就点个赞呗~