PAT 乙级 1059 C语言竞赛 (20 分)

C 语言竞赛是浙江大学计算机学院主持的一个欢乐的竞赛。既然竞赛主旨是为了好玩，颁奖规则也就制定得很滑稽：

• 0、冠军将赢得一份“神秘大奖”（比如很巨大的一本学生研究论文集……）。
• 1、排名为素数的学生将赢得最好的奖品 —— 小黄人玩偶！
• 2、其他人将得到巧克力。

给定比赛的最终排名以及一系列参赛者的 ID，你要给出这些参赛者应该获得的奖品。

输入格式：

输入第一行给出一个正整数 N（≤104），是参赛者人数。随后 N 行给出最终排名，每行按排名顺序给出一位参赛者的 ID（4 位数字组成）。接下来给出一个正整数 K 以及 K 个需要查询的 ID。

输出格式：

对每个要查询的 ID，在一行中输出 ID: 奖品，其中奖品或者是 Mystery Award（神秘大奖）、或者是 Minion（小黄人）、或者是 Chocolate（巧克力）。如果所查 ID 根本不在排名里，打印 Are you kidding?（耍我呢？）。如果该 ID 已经查过了（即奖品已经领过了），打印 ID: Checked（不能多吃多占）。

输入样例：

6
1111
6666
8888
1234
5555
0001
6
8888
0001
1111
2222
8888
2222

输出样例：

8888: Minion
0001: Chocolate
1111: Mystery Award
2222: Are you kidding?
8888: Checked
2222: Are you kidding?

分析：

使用一个map存储每个人对应的礼物，再用一个check数组检查某个id的人是否已经领取过礼物。

代码：

#include <iostream>
#include <algorithm>
#include <string>
#include <map>
#include <cmath>

using namespace std;

// 判断是不是素数
bool isPrime(int a) {
    if (a <= 1) return false;
    int sqrt_ = sqrt((double)a);
    for (int i = 2; i <= sqrt_; i++) {
        if (a % i == 0) {
            return false;
        }
    }
    return true;
}

int main() {
    int N;
    cin >> N;
    map<string, string> mp;
    int check[10000] = { 0 };
    string prize, id;
    for (int i = 0; i < N; i++) {
        cin >> id;
        
        // 为每个人分配礼物
        if (i < 2) {
            prize = "Mystery Award";
        } else if (isPrime(i + 1)) {
            prize = "Minion";
        } else {
            prize = "Chocolate";
        }
        mp[id] = prize;
    }

    cin >> N;
    auto end = mp.end();
    for (int i = 0; i < N; i++) {
        cin >> id;
        int temp = stoi(id);
        if (mp.find(id) != end && !check[temp]) {
            cout << id << ": " << mp[id] << endl;
            check[temp] = 1;  // 标记已经领取过礼物了
        } else if (mp.find(id) != end && check[temp]) {
            cout << id << ": Checked" << endl; 
        } else {
            cout << id << ": Are you kidding?" << endl;
        }
    }
    
    return 0;
}