PTA 1059 c语言竞赛

1059 C语言竞赛 (20 分)
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（不能多吃多占）。

#include<bits/stdc++.h>
struct A{
	int flag;
	int num;
};
bool su(int i){
	for(int j=2;j<=sqrt(i);j++){
		if(i%j==0)
			return false;
	}
	return true;
}
using namespace std;
int main(){
	int n,m;
	cin>>n;
	A a[10001];
	for(int i=1;i<=n;i++)
		{
		cin>>a[i].num;
		a[i].flag=0;}
	cin>>m;
	while(m>0){
		int x;
		cin>>x;
		int z=0;
		for(int i=1;i<=n;i++){
			if(a[i].num==x){
				if(a[i].flag==0)
					{	
						if(i==1){
							printf("%04d",x);
						cout<<": Mystery Award"<<endl;
						}
						else if(su(i)){
								printf("%04d",x);
						cout<<": Minion"<<endl;
						}
						else{
								printf("%04d",x);
						cout<<": Chocolate"<<endl;
						}
						a[i].flag=1;
					}
				else{
						printf("%04d",x);
					cout<<": Checked"<<endl;
				}
				z=1;
			}
		}
		if(z==0)
			{
			printf("%04d",x);
		cout<<": Are you kidding?"<<endl;}
        m--;
	}
	return 0;
}