PAT-B1059 C语言竞赛（20 分）

1059 C语言竞赛（20 分）

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

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

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

输入格式：

输入第一行给出一个正整数 N（≤10​4​​），是参赛者人数。随后 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?

分析：

先计算出N以内（包括N）的素数，这里使用线性筛法。保存ID的数组下标加1就是某个ID的排名。

代码：

#include <cstdio>
#include <string>
#include <vector>

using namespace std;

int main()
{
    int N, K, i, j;
    scanf("%d", &N);
    vector<int> prime;
    bool isNotPrime[N + 2] = {false};//将所有的数设置为是素数
    for(i = 2; i < N + 2; i++)//
    {   //计算N以内的素数
        if(isNotPrime[i] == false)
            prime.push_back(i);
        //筛去以i为最大因数的合数
        for(j = 0; j < prime.size() && i * prime[j] <= N + 2; j++)
        {
            isNotPrime[i * prime[j]] = true;
            if(i % prime[j] == 0)//找到了i的最小素因数
                break;
        }
    }
    //printf("%d以内的素数个数：%d\n", N, prime.size());
    int id[N], target;
    bool check[N] = {false};
    for(i = 0; i < N; i++)
        scanf("%d", &id[i]);
    scanf("%d", &K);
    for(i = 0; i < K; i++)
    {
        scanf("%d", &target);
        bool flag = false;
        for(j = 0; j < N; j++)
        {
            if(target == id[j])
            {
                flag = true;//匹配成功
                if(check[j] == true)
                    printf("%04d: Checked\n", target);
                else if(j == 0)
                    printf("%04d: Mystery Award\n", target);
                else if(!isNotPrime[j + 1])//排名为素数
                    printf("%04d: Minion\n", target);
                else
                    printf("%04d: Chocolate\n", target);
                check[j] = true;
                break;
            }
        }
        if(flag == false)
            printf("%04d: Are you kidding?\n", target);
    }
    return 0;
}