PAT乙级刷题/1059 C语言竞赛/C++实现

一、题目描述 

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?

二、题解代码以及提交截图

#include <iostream>
#include <map>

bool IsPrime(int num){
    for(int i = 2;i * i < num;i++){
        if(num % i == 0){
            return false;
        }
    }
    if(num == 2){
        return true;
    }
    return true;
}
int main()
{
    int N;
    std::cin >> N;
    int flag[10000] = {0};
    std::map<std::string,int> list;
    for (int i = 1; i <= N; i++) {
        std::string str1;
        std::cin >> str1;
        list.insert({str1, i});
    }
    int K;
    std::cin >> K;
    for (int j = 0; j < K; j++) {
        std::string str2;
        std::cin >> str2;
        int str2_num = std::stoi(str2);
        flag[str2_num]++;
        auto search = list.find(str2);
        if(search != list.end() && flag[str2_num] > 1){
            std::cout << str2 << ": " << "Checked";
            std::cout << std::endl;
        }
        else{
            if(search != list.end()){
                for(auto & itr : list){
                    if(itr.first == str2){
                        if(itr.second == 1){
                            std::cout << str2 << ": " << "Mystery Award";
                            std::cout << std::endl;
                        }
                        else if(IsPrime(itr.second)){
                            std::cout << str2 << ": " << "Minion";
                            std::cout << std::endl;
                        }
                        else{
                            std::cout << str2 << ": " << "Chocolate";
                            std::cout << std::endl;
                        }
                    }
                }
            }
            else{
                std::cout << str2 << ": " << "Are you kidding?";
                std::cout << std::endl;
            }
        }
    }
}