PTA 乙级 1072 开学寄语 (20分) C++

 测试点2：问题物品四位输出的零位补齐

思路：输入即检测，检测出问题即输出，看代码吧

#include<iostream>
#include<vector>

using namespace std;

int main() {
    int n = 0, m = 0;
    /*问题学生,问题物品总数*/
    int pro_stu = 0, pro_obj = 0;
    /*问题物品标记*/
    int obj[10000] = { 0 };
    /*姓名输出标志位*/
    int flag = 1;
    cin >> n >> m;
    for (int i = 0; i < m; ++i) {        //标记问题物品
        int tmp = 0;
        cin >> tmp;
        obj[tmp] = 1;
    }
    for (int i = 0; i < n; ++i) {
        string name;
        int k = 0;
        cin >> name >> k;
        for (int j = 0; j < k; ++j) {
            int tmp = 0;
            cin >> tmp;
            if (obj[tmp]) {
                if (flag) {                //没有输出过姓名
                    cout << name << ":";
                    flag = 0;
                    pro_stu++;
                }
                printf(" %04d", tmp);    //注意物品不满四位的0补齐
                pro_obj++;
            }
        }
        if(!flag) cout << endl;            //输出过问题学生则回车换行
        flag = 1;
    }
    cout << pro_stu << ' ' << pro_obj;
    return 0;
}