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 <iostream>
#include <stdio.h>
#include <cmath>
using namespace std;

bool Isprime(int a)
{
    bool flag = true;
    if ( a <= 1){
        flag = false;
    }else if( a == 2){
        flag = true;
        }else{
            if ( a % 2 == 0){
                flag = false;
            }else{
                for ( int i = 3 ; i <= sqrt(a) ; i += 2){
                    if ( a % 2 == 0){
                        flag = false;
                        break;
                    }
                } 
            }
        }
    return flag; 
}

int main()
{
    int N;
    cin>>N;
    int order[10000] = {0};
    for ( int i = 0 ; i < N ; i++){
        int id;
        cin>>id;
        order[id] = i+1;
    }

    int n;
    cin>>n;
    int check[n];
    int count = 0;
    for ( int i = 0 ; i < n ; i++){
        int search;
        cin>>search;
        if (order[search] == 0){
            printf("%4d",search);
            cout<<": Are you kidding?"<<endl;
        }else if (order[search] == -1){
                printf("%04d",search);
                cout<<": Checked"<<endl;
            }else{
                if ( order[search] == 1){
                    printf("%04d",search);
                    cout<<": Mystery Award"<<endl;
                }else{
                    if (Isprime(order[search]) == true){
                        printf("%04d",search);
                        cout<<": Minion"<<endl;
                    }else{
                        printf("%04d",search);
                        cout<<": Chocolate"<<endl;
                    }
                }
            order[search] = -1;
            }
        }

    return 0;
}