PTA乙级1072，开学寄语

#include<bits/stdc++.h>
using namespace std;

int a[9999]={0};//先创建一个用于判断是否是违禁物品的数组
int main(){
    int n,m,temp,k,snum=0,fnum=0,x;
    scanf("%d %d",&n,&m);//输入学生总人数和违禁物品数
    for(int i=0;i<m;i++){
        scanf("%d",&temp);//输入所有违禁物品编号
        a[temp]=1;//以违禁物品编号为索引的数组值相应记为1
    }

    for(int i=0;i<n;i++){
        char name[10];
        bool flag=false;//用于判断学生姓名是否已经输出过
        cin>>name>>k;//输入学生姓名和学生携带物品数量

        for(int j=0;j<k;j++){//输入学生携带物品编号
            cin>>x;
            if(a[x]==1){//是违禁物品
                if(!flag){//学生姓名尚未输出过
                    cout<<name<<":";
                    flag=true;//学生姓名已经输出过
                }
                printf(" %04d", x);
                fnum++;
            }
        }

        if(flag){//已完成对一名学生的所有操作，换行
            cout<<"\n";
            snum++;
        }
    }
    printf("%d %d\n", snum, fnum);

    return 0;
}