PAT乙 1059. C语言竞赛

1059. C语言竞赛(20)

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?

求解过程:

通过用空间换时间的方式来提高效率，首先开一个10001大小的数组Arr，初始化为0，然后每输入一个ID，就令Arr[ID]=i+1，其中i是输入的顺序，用一个for循环即可完成。剩下的就按部就班，稍加注意输出的时候要加上%04d以防止出现ID为0001的输出为1.

程序：

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

int Stu[100001] = {0};		// 以ID为下标,排名为值
int visited[100001] = {0};	// 记录访问与否
int isPrimeNum(int x)		// 求是否为素数
{
	for (int i = 2; i <= sqrt(x); i++)
		if (x % i == 0)		// 一旦出现有一个数可以整除x,就返回0,不是素数
			return 0;
	return 1;				// 否则是素数
}	

int main(int argc, char const *argv[])
{
	int N, K, ID;
	cin >> N;
	for (int i = 0; i < N; i++)		// 初始化数组
	{
		cin >> ID;
		Stu[ID] = i+1;				// 排名从1开始。
	}
	cin >> K;
	for (int i = 0; i < K; i++)
	{
		cin >> ID;
		if (!Stu[ID])				// 如果Stu[ID]为0，说明不存在
			printf("%04d: Are you kidding?\n", ID);
		else						// 如果存在
		{
			if (visited[ID])		// 如果已访问
				printf("%04d: Checked\n", ID);
			else					// 如果未访问
			{
				if (Stu[ID] == 1)
					printf("%04d: Mystery Award\n", ID);
				else if (isPrimeNum(Stu[ID]))
					printf("%04d: Minion\n", ID);
				else
					printf("%04d: Chocolate\n", ID);
			}
			visited[ID] = 1;		// 将该点置为1，表示已访问
		}
	}
	return 0;
}