1059. C语言竞赛(20)

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?

#include<iostream>
#include<cmath>
#include<cstdio>
using namespace std;
int is_prime(int num){
	int flag=1;
	for (int i=2;i<=sqrt(num);i++){
		if (num%i==0)
		{
			flag = 0;
			break;
		}
	}
	return flag;
}
int main(){
	int flag[10000],n,id,k;//flag存在为1，不存在为0，访问过为-1； 
	cin>>n;
	int result[n+1];	//result存排名 
	for (int i=1;i<=n;i++)		
	{
		cin>>id;
		result[id] = i;
		flag[id] = 1;	
	}
	cin>>k;
	for (int i=0;i<k;i++){
		int temp;
		cin>>temp;
		if (flag[temp]==1){		//存在 
			if (result[temp]==1){
				printf("%04d: Mystery Award\n",temp); 
				flag[temp] = -1;	//设置标志 
				}
			else if (is_prime(result[temp])){
				printf("%04d: Minion\n",temp); 
				flag[temp] = -1;
				} 
			else { 
				printf("%04d: Chocolate\n",temp); 
				flag[temp] = -1;
				}						
		}
		else if (flag[temp]==-1)	//访问过 
			printf("%04d: Checked\n",temp); 
		else 						//不存在 
			printf("%04d: Are you kidding?\n",temp); 
	}
	return 0;
}