PAT Basic 1059 C语言竞赛 (20 分)

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

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

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

输入格式：

输入第一行给出一个正整数

输出格式：

对每个要查询的 ID，在一行中输出 ID: 奖品，其中奖品或者是 Mystery Award（神秘大奖）、或者是 Minion（小黄人）、或者是 Chocolate（巧克力）。如果所查 ID 根本不在排名里，打印 Are you kidding?（耍我呢？）。如果该 ID 已经查过了（即奖品已经领过了），打印 ID: Checked（不能多吃多占）。

输入样例：

输出样例：

8888: Minion
0001: Chocolate
1111: Mystery Award
2222: Are you kidding?
8888: Checked
2222: Are you kidding?

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
int award[100000]={0};//1,2,3为3个奖品
bool isPrime(int a){
    if(a==2||a==3) return true;
    for(int i=2;i<=sqrt(a);i++){
        if(a%i==0) return false;
    }
    return true;
}
int main()
{
    int n;cin>>n;
    int a[n];
    for(int i=0;i<n;i++) cin>>a[i];
    for(int i=0;i<n;i++) award[a[i]]=3;
    award[a[0]]=1;
    for(int i=1;i<n;i++) if(isPrime(i+1)) award[a[i]]=2;
    int m,tmp;
    cin>>m;
    while(m--){
        cin>>tmp;
        if(award[tmp]==1) {
            printf("%04d: Mystery Award
",tmp);
            award[tmp]=4;
        }
        else if(award[tmp]==2) {
            printf("%04d: Minion
",tmp);
            award[tmp]=4;
        }
        else if(award[tmp]==3) {
            printf("%04d: Chocolate
",tmp);
            award[tmp]=4;
        }
        else if(award[tmp]==4) printf("%04d: Checked
",tmp);
        else printf("%04d: Are you kidding?
",tmp);
    }
    system("pause");
    return 0;
}