pat乙级1059C语言

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?

思路

定义一个函数判断素数，然后将输入的序号存放在一个数组中记录排名，然后对需要输出的进行比对，第一和素数还有其他以及多次出现的要判断，这里我在一次过后对这个序号的排名赋值-1，第二次出现就可以找出。

#include
#include

int is_sushu(int a)//判断是否是素数
{
    for(int i=2;i<=sqrt(a);i++)
    {
        if(a%i==0) return 0;
    }
    return 1;
}

int main()
{
    int number[10000]={0};//用来存放排名
    int n,temp,k;//n是初始的排名个数
    scanf("%d",&n);
    for(int i=1;i<=n;i++)//输入排名
    {
        scanf("%d",&temp);
        number[temp]=i;//给有名次的赋值
    }
    scanf("%d",&k);//要验证的id个数
    for(int i=1;i<=k;i++)
    {
        scanf("%d",&temp);
        if(number[temp]==0)
        {
            printf("%04d: Are you kidding?\n",temp);//不存在
        }
        else
        {
            if(number[temp]==-1)
            printf("%04d: Checked\n",temp);//领过了
            else if(number[temp]==1)
            printf("%04d: Mystery Award\n",temp);//第一
            else if(is_sushu(number[temp]))
            printf("%04d: Minion\n",temp);//小黄人
            else
            printf("%04d: Chocolate\n",temp);//巧克力
            number[temp]=-1;
        }
    }
}