【PAT乙级】1059 C语言竞赛

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?

个人理解

这题可能唯一有点麻烦的就是素数的处理，我采用的方法是用时间复杂度为O(n)的素数筛法生成10^4以内的素数表，然后在查询排名时直接查表即可。

代码实现

#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <algorithm>
#include <iostream>
#define ll long long
#define ep 1e-5
#define INF 0x7FFFFFFF

int const maxn = 10010;

using namespace std;

//素数筛法 prime中存储所有素数，is_prime保存下标数组是否是素数
int prime[maxn];
bool is_prime[maxn];

int get_prime(int n)
{
    int cnt = 0;
    for (int i = 0; i <= n; i ++) {
        is_prime[i] = true;
    }
    is_prime[0] = is_prime[1] = false;
    for (int i = 2; i <= n; i ++) {
        if (is_prime[i]) {
            prime[cnt++] = i;
            for (int j = 2; j*i <= n; j ++) {
                is_prime[i*j] = false;
            }
        }
    }
    return cnt;
}

int main() {
    //初始化
    int students[maxn], n;
    bool is_checked[maxn];
    memset(students, 0, sizeof(students));
    memset(is_checked, 0, sizeof(is_checked));
    
    //输入
    cin >> n;
    for (int i = 1; i <= n; i ++) {
        int id;
        cin >> id;
        students[id] = i;
    }
    
    //生成素数表
    get_prime(maxn);
    
    //对排名进行判断并输出结果
    int k;
    cin >> k;
    for (int i = 0; i < k; i ++) {
        int id;
        cin >> id;
        printf("%04d: ",id);
        if (students[id] == 0) {
            cout << "Are you kidding?" << endl;
        }
        else if (is_checked[id]) {
            cout << "Checked" << endl;
        }
        else if (students[id] == 1) {
            cout << "Mystery Award" << endl;
        }
        else if (is_prime[students[id]]) {
            cout << "Minion" << endl;
        }
        else {
            cout << "Chocolate" << endl;
        }
        is_checked[id] = true;
    }
    
    return 0;
}

总结

学习不息，继续加油