PAT B1059 C语言竞赛 (20分)

题目链接：https://pintia.cn/problem-sets/994805260223102976/problems/994805269828059136

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

0、冠军将赢得一份“神秘大奖”（比如很巨大的一本学生研究论文集……）。
1、排名为素数的学生将赢得最好的奖品 —— 小黄人玩偶！
2、其他人将得到巧克力。
给定比赛的最终排名以及一系列参赛者的 ID，你要给出这些参赛者应该获得的奖品。

输入
输入第一行给出一个正整数 N（≤10^​4​​ ），是参赛者人数。随后 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 <cstdio>

const int maxn = 100010;
int hash[maxn] = {0}, rank[maxn] = {0};

bool prime(int x) {
	if(x <= 1)
		return false;
	for(int i = 2; i * i <= x; i++)
		if(x % i == 0)
			return false;
	return true;
}

int main() {
	int n, k, id;
	scanf("%d", &n);
	for(int i = 0; i < n; i++) {
		scanf("%d", &id);
		rank[id] = i + 1;
		hash[id] = 2;
	}
	scanf("%d", &k);
	for(int i = 0; i < k; i++) {
		scanf("%d", &id);
		printf("%04d: ", id);
		if(hash[id] == 0)
			printf("Are you kidding?\n");
		else {
			if(hash[id] == 1)
				printf("Checked\n");
			else {
				if(rank[id] == 1)
					printf("Mystery Award\n");
				else if(prime(rank[id]))
					printf("Minion\n");
				else
					printf("Chocolate\n");
				hash[id] = 1;
			}
		}
	}
	return 0;
}