1059. C语言竞赛(20)

1059. C语言竞赛(20)
时间限制 200 ms 内存限制 65536 kB 代码长度限制 8000 B
判题程序 Standard 作者 CHEN, Yue

C语言竞赛是浙江大学计算机学院主持的一个欢乐的竞赛。既然竞赛主旨是为了好玩，颁奖规则也就制定得很滑稽：
0. 冠军将赢得一份“神秘大奖”（比如很巨大的一本学生研究论文集……）。
1. 排名为素数的学生将赢得最好的奖品 —— 小黄人玩偶！
2. 其他人将得到巧克力。
给定比赛的最终排名以及一系列参赛者的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?

#define _CRT_SECURE_NO_WARNINGS
#include <algorithm>
#include <iostream>
#include <cstring>
#include <iomanip>
#include <vector>
#include <string>
#include <cfloat>
#include <queue>
#include <map>

using namespace std;
map<string, int>Rank;
const int MaxN = 10010;
enum TYPE {PRIME,FRIST,CHCO,CHEC,NONE};
bool isChecked[MaxN];

bool isPrime(int n)
{
    if (n <= 1) return false;
    if (n == 2) return true;
    int qrt = sqrt(n);
    for (int i = 2; i <= qrt; ++i)
    {
        if (n % i == 0)return false;
    }

    return true;
}

TYPE Judge(string id)
{
    if (Rank.count(id))
    {
        int rank = Rank[id];
        if (isChecked[rank] == false)
        {
            isChecked[rank] = true;
            if (rank == 1)return FRIST;
            if (isPrime(rank))return PRIME;
            return CHCO;
        }
        return CHEC;
    }
    return NONE;
}


int main()
{
#ifdef _DEBUG
    freopen("data.txt", "r+", stdin);
#endif // _DEBUG
    std::ios::sync_with_stdio(false);

    int n;
    string str;
    cin >> n;
    for(int i=1;i<=n;++i)
    {
        cin >> str;
        Rank[str] = i;
    }

    cin >> n;
    for (int i = 0; i < n; ++i)
    {
        cin >> str;
        TYPE type = Judge(str);
        switch (type)
        {
        case FRIST:
            cout << str << ": Mystery Award\n";
            break;
        case PRIME:
            cout << str << ": Minion\n";
            break;
        case CHCO:
            cout << str << ": Chocolate\n";
            break;
        case NONE:
            cout << str << ": Are you kidding?\n";
            break;
        case CHEC:
            cout << str << ": Checked\n";
            break;
        default:
            break;
        }
    }
    return 0;
}