[PAT乙级]1059. C语言竞赛(20)

1059. C语言竞赛(20)

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

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

给定比赛的最终排名以及一系列参赛者的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?

思路：

• 这道题主要考察数组应用

代码：

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
bool isPrimeNum(int i){
    for(int j=2; j*j<=i; j++){//j*j很重要，减少运行时间
        if(i%j == 0){
            return false;
        }
    }
    return true;
}
int main()
{
    int N;
    cin >> N;
    vector<int> arr(10001);
    for(int i=1; i<=N; i++){//记录每个id排名
        int id;
        cin >> id;
        arr[id] = i;
    }
    int K;
    cin >> K;
    vector<int> aa(10001);//记录每个id有没有领过奖品
    vector<int> res(K);
    for(int i=0; i<K; i++){
        cin >> res[i];//输入要查找的Id
    }
    for(int i=0; i<res.size(); i++){
        printf("%04d", res[i]);
        printf(": ");
        if(arr[res[i]] == 0){//如果id没有排名
            cout << "Are you kidding?" << endl;
            continue;
        }
        if(aa[res[i]] == 1){//如果id领过奖品
            cout << "Checked" << endl;
            continue;
        }
        if(aa[res[i]] == 0){//如果id有没有领过奖品
            if(arr[res[i]] == 1){
                cout << "Mystery Award" << endl;

            }
            else if(isPrimeNum(arr[res[i]])){
                cout << "Minion" << endl;
            }
            else
                cout << "Chocolate" << endl;
            aa[res[i]] = 1;
        }
    }
    return 0;
}