PAT乙级真题 1059 C语言竞赛 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?

思路

将id视为string，用STL库中map存储id与排名的对照关系，由姓名即可直接索引出排名。

用一个bool数组记录每个排名的选手是否领过物品。

map判断key是否存在有两种方法：

1、使用find()，返回值是迭代器位置，若返回.end()说明不存在。

if (mymap.find(key) == mymap.end())
    cout << "没有这个key" << endl;

2、使用count()，返回值是key出现的次数，即0或1。

if (mymap.count(key) == 0)
    cout << "没有这个key" << endl;

代码

#include <iostream>
#include <vector>
#include <cmath>
#include <map>
using namespace std;

bool isPrime(int n){
    //用pow(n, 0.5)效率低下，sqrt效率高
    for (int i=2; i<=sqrt(n); i++){
        if (n%i==0){
            return false;
        }
    }
    return true;
}

int main(){
    int n;
    cin >> n;
    map<string, int> ids;
    vector<bool> checked(n);
    for (int i=0; i<n; i++){
        string s;
        cin >> s;
        ids[s] = i + 1;
    }

    int k;
    cin >> k;
    for (int i=0; i<k; i++){
        string s;
        cin >> s;
        cout << s << ": ";
        if (ids.find(s)==ids.end()){
            cout << "Are you kidding?" << endl;
        }
        else{
            int rank = ids[s];
            if (checked[rank-1]){
                cout << "Checked" << endl;
            }
            else {
                if (rank==1){
                    cout << "Mystery Award" << endl;
                }
                else if(isPrime(rank)){
                    cout << "Minion" << endl;
                }
                else{
                    cout << "Chocolate" << endl;
                }
                checked[rank-1] = true;
            }
        }
    }
    
    return 0;
}