PAT-B 1059. C语言竞赛

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

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

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

输入格式：

输入第一行给出一个正整数N（<=10000），是参赛者人数。随后N行给出最终排名，每行按排名顺序给出一位参赛者的ID（4位数字组成）。接下来给出一个正整数K以及K个需要查询的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<stdio.h>
#include<math.h>
#define MAX 10000
int isPrime(int n);
int rank[MAX];
int flag[MAX];
int main()
{
    int n,k,i=1,ID;
    scanf("%d",&n);
    while(i<=n)
    {
        scanf("%d",&ID);
        rank[ID]=i;
        i++;
    }
    scanf("%d",&k);
    while(k--)
    {
        scanf("%d",&ID);
        if(flag[ID]==0)
        {
            if(rank[ID]==1)
            {
                 printf("%04d: Mystery Award\n",ID);
            }
            else if(rank[ID]==0)
            {
                printf("%04d: Are you kidding?\n",ID);
            }
            else if(isPrime(rank[ID]))
            {
                 printf("%04d: Minion\n",ID);

            }
            else
            {
                 printf("%04d: Chocolate\n",ID);
            }
            if(rank[ID]!=0)
                flag[ID]=1;
        }
        else
        {
            printf("%04d: Checked\n",ID);
        }

    }
    return 0;
}
int isPrime(int n)
{
    int i;
    if(n<2)
        return 0;
    else if(n==2)
        return 1;
    else
    {
        for(i=2;i<=sqrt(n);i++)
        {
            if(n%i==0)
                return 0;
        }
    }
    return 1;
}