PAT --- 1059.C语言竞赛 （20 分）

1059 C语言竞赛 （20 分）

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

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

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

输入格式：

输入第一行给出一个正整数 $N$（≤104\le 10^4≤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?

思路

使用Comp结构，保存编号和是否查询过，true为未查询，false为已查询。
使用isPrime判断是否为素数。
注意for循环中i由1开始遍历，便于计算素数。
唯一不足是使用for循环嵌套，虽然相对柳神的代码节省了空间，但时间复杂度较高。
具体如注释

代码

#include <iostream>
#include <vector>
#include <cmath>
using namespace std;

struct Compe					//使用id保存编号
{
	int	id;
	bool isSearch = true;		//默认true 查询后置为false
};

bool isPrime(int a) {			//判断是否为素数
	if (a <= 1) return false;
	int nSqrt = sqrt((double)a);
	for (int i = 2; i <= nSqrt; i++) {
		if (a % i == 0)
			return false;
	}
	return true;
}

int main()
{
	int n, k, nId;
	cin >> n;
	vector<Compe> vCompe(n + 1);
	for (int i = 1; i <= n; i++) {
		cin >> nId;
		vCompe[i].id = nId;
	}
	cin >> k;
	for (int i = 1; i <= k; i++) {
		cin >> nId;
		bool flag = true;						// 查到置为false
		for (int j = 1; j <= n; j++) {
			if (vCompe[j].id == nId) {
				flag = false;
				if (j == 1) {					//如果是第一名 神秘大奖
					if (vCompe[j].isSearch) {
						printf("%04d: Mystery Award\n", nId);
						vCompe[j].isSearch = false;	//查到之后置为false
					}
					else
						printf("%04d: Checked\n", nId);
				}
				else if (isPrime(j)) {			//如果是素数 小黄人
					if (vCompe[j].isSearch) {
						printf("%04d: Minion\n", nId);
						vCompe[j].isSearch = false;
					}
					else
						printf("%04d: Checked\n", nId);
				}
				else {							//其他  巧克力
					if (vCompe[j].isSearch) {
						printf("%04d: Chocolate\n", nId);
						vCompe[j].isSearch = false;
					}
					else
						printf("%04d: Checked\n", nId);
				}
			}
		}
		if (flag)								//遍历之后未找到
			printf("%04d: Are you kidding?\n", nId);
	}
    return 0;
}