1059. C语言竞赛(20)

时间限制
200 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CHEN, Yue

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

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

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

输入格式：

输入第一行给出一个正整数N（<=10000），是参赛者人数。随后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?

解题思路

本题并不难，但是因为自己把判断素数的函数写错了，导致测试点3一直不能通过，搞了好久才找到错误，希望下次能汲取教训。

#include<iostream>
#include<map>
#include<string>
#include<list>
#include<cmath>
using namespace std;

bool isPrime(int num) {

    if (num < 2)//素数>=2
    {
        return false;
    }
    for (int j = 2; j <= sqrt(num); j++) {
        if (num % j == 0) return false;
    }
    return true;
}

int main() {
    int num;
    string id;
    map<string,int> m_map;
    map<string, int> get_map;
    list<string> result;

    cin >> num;
    for (int i = 1; i <= num;i++) {
        cin >> id;
        m_map[id] = i;
    }
    cin >> num;
    for (int i = 0; i < num; i++)
    {
        cin >> id;
        if (get_map.count(id) != 0)
        {
            cout << id << ": Checked" << endl;
            continue;
        }
        if (m_map.count(id) == 0){
            cout << id << ": Are you kidding?" << endl;
            continue;
        }

        if (m_map[id] == 1)
        {
            cout << id << ": Mystery Award" << endl;
            get_map[id] = m_map[id];
            m_map.erase(id);
            continue;
        }

        if (isPrime(m_map[id]) == true)
        {
            cout << id << ": Minion" << endl;
            get_map[id] = m_map[id];
            m_map.erase(id);
            continue;
        }

        cout << id << ": Chocolate" << endl;
        get_map[id] = m_map[id];
        m_map.erase(id);

    }

    return 0;
}