PAT 1059 C语言竞赛（20）（代码+思路）

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（不能多吃多占）。

输入样例：

输出样例：

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

PS:

这里我用了map<string，struct>，结构体中string存放奖品，int判断ID是否存在。

过程：

输入->由于按排名输入，所以我们可以直接根据条件在输入的时候把奖品（prize）赋值，同时（key置1）

查找->通过map索引输入的字符串，首先判断key，若为1，输出奖品，同时将奖品置为“Checked”（因为只能访问一次）

注意：尽可能降低时间复杂度

#include<iostream>
#include<string>
#include<cmath>
#include<map>
using namespace std;
struct m {
	string prize;		//存放奖品
	int key = 0;	//ID是否存在
};
map<string, m> ranking;
int pd(int a) {
	for (int i = 2; i <= sqrt(a); i++)
		if (a%i == 0)
			return 0;
	return 1;
}
int main() {
	int n, flag;
	string str;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> str;
		ranking[str].key = 1;		
		if (i == 1)
			ranking[str].prize = "Mystery Award";
		else if (pd(i))
			ranking[str].prize = "Minion";
		else
			ranking[str].prize = "Chocolate";
	}
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> str;
		if (ranking[str].key) {
			cout << str << ": " << ranking[str].prize << endl;
			ranking[str].prize = "Checked";
		}
		else
			cout << str << ": Are you kidding?" << endl;
	}
	return 0;
}