【PAT Basic Level】1072 开学寄语

专栏：PAT乙级刷题笔记

题目

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

输入格式

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

分析

设置一个违禁品标记数组prohib[10000]，当prohib[ID] = 1表示改编号物品是违禁品。尽管违禁品数量最多为6不过空间浪费就浪费吧。
t用于标记该学生是否携带违禁品。
那么当该学生携带了违禁品，且t依旧等于0时，就可以输出ta的名字了。同时违纪学生数加1。
应该算得上是简洁的代码了

代码

#include<iostream>
#include<math.h>
using namespace std;

int prohib[10000] = {0};
int main(){
	int N,M,K,t,ID,c1 = 0,c2 = 0;
	cin>>N>>M;
	string s;
	for(int i = 0;i < M;++i){
		cin>>ID;prohib[ID] = 1;
	}
	for(int i = 0;i < N;++i){
		cin>>s>>K;
        t = 0;
		for(int j = 0;j < K;++j){
			cin>>ID;
			if(prohib[ID] == 1){
				if(t == 0){
					cout<<s<<":";
					++c1;//违纪学生数+1
				}
				printf(" %04d",ID);
				t = 1;
				++c2;//违禁品+1
			}
		}
		if(t == 1)
			cout<<endl;
	}
	cout<<c1<<' '<<c2;
}