PAT-B-1059-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?

Code

#include <iostream>
#include <set>
#include <map>
#include <string>
using namespace std;

//素数的判定
bool isPrime(int n) {
	if (n <= 1)
		return false;
	for (int i = 2;i * i <= n;i++) {
		if (n % i == 0)return false;
	}
	return true;
}

int main() {
	int n, k, rank = 1;
	string temp,cpId;//单独记录冠军的id
	map<string, int> m;
	scanf("%d", &n);
	

	for (int i = 0;i < n;i++) {
		cin>>temp;
		if (i == 0)
			cpId = temp;
		m.insert(make_pair(temp,i));
	}

	scanf("%d", &n);
	set<string> checked;//放置查询过的id
	set<string>::iterator it;
	map<string, int>::iterator mit;
	while (n--) {
		cin >> temp;
		mit = m.find(temp);
		if (mit != m.end()) {
			it = checked.find(temp);
			if (it != checked.end()) {
				//已经领过奖了
				printf("%s: Checked\n", temp.c_str());
			}else {
				checked.insert(temp);
				if (temp == cpId) {
					printf("%s: Mystery Award\n", temp.c_str());
				}
				else {
					if (isPrime(mit->second+1)) {
						printf("%s: Minion\n", temp.c_str());

					}else {
						printf("%s: Chocolate\n", temp.c_str());
					}

				}
			}
			
		}else {

			printf("%s: Are you kidding?\n", temp.c_str());
			
		}

	}

	return 0;
}