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 <string>
#include <vector>
#include <math.h>
using namespace std;

int change(int x){
if(x<10)cout<<"000"<<x;
else if(x<100)cout<<"00"<<x;
else if(x<1000)cout<<0<<x;
else cout<<x;
return 0;
}

int prime(int n){
for(int k=2;k<=sqrt(n);k++) {
if((n%k)==0)return 0;
}
return 1;
}

struct player{
int ID;
int check;
};

int main(){
int N,K,i,j,a,m=0;
vector<player> v;
cin>>N;
for(i=0;i<N;i++){
player s;
cin>>a;
s.ID=a;
s.check=1;
v.push_back(s);
}
cin>>K;
for(i=0;i<K;i++){
cin>>a;
for(j=0;j<v.size();j++){
m=0;
if(v[j].ID==a){
if(v[j].check==0){change(a);cout<<": Checked"<<endl;}
else if(j==0){change(a);cout<<": Mystery Award"<<endl;}
else if(prime(j+1)==1){change(a);cout<<": Minion"<<endl;}
else {change(a);cout<<": Chocolate"<<endl;}
v[j].check=0;
m=1;
break;
}
}
if(m==0){change(a);cout<<": Are you kidding?"<<endl;}
}
return 0;
}