1059. C语言竞赛(20)

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

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

给定比赛的最终排名以及一系列参赛者的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<cstdio>
#include<cmath>
using namespace std;

int N,K,stu[10005]={0};     //刚开始置为0
int judge(int v){           //判断素数
        for(int j=2;j<=sqrt(v);j++){
            if(v%j==0) return 0;
            else if(j>sqrt(v))  return 1;
    }
}
int main(){
    scanf("%d",&N);
    //以id为下标，把输入的顺序即排名为数组元素的值存储起来，这样做的一个好处是以散列方式可以快速知道是否在数组里
    for(int i=1;i<=N;i++){  
        int temp;
        scanf("%d",&temp);
        stu[temp]=i;
    }
    scanf("%d",&K);
    for(int i=0;i<K;i++){
        int temp;
        scanf("%d",&temp);
        if(stu[temp]==0)    //如果没有这个id，则查找的时候以这个id为下标的元素为0
            printf("%04d: Are you kidding?\n",temp);
        else if(stu[temp]==-1)//如果元素值为-1则代表这个元素已经被查过一遍了，因为查过一遍才会将数组的值置为-1
            printf("%04d: Checked\n",temp);
        else if(stu[temp]==1){  //如果数组元素为1说明排名为1，则输出神秘大奖，并将元素值置为-1，以防下一次查到
            printf("%04d: Mystery Award\n",temp);
            stu[temp]=-1;
        }
        else if (judge(stu[temp])){//判断是否为素数，如果是，函数返回1，并输出小黄人，否则返回0，同样以防下次查到将元素值置为-1
            printf("%04d: Minion\n",temp);
            stu[temp]=-1;
        }
        else{//如果不是以上情况，则输出巧克力并置为-1，以防下次查到
            printf("%04d: Chocolate\n",temp);
            stu[temp]=-1;
        } 
    }
    return 0;
}