Basic Level 1059 C語言競賽 (20分)

題目

C 語言競賽是浙江大學計算機學院主持的一個歡樂的競賽。既然競賽主旨是為了好玩，頒獎規則也就制定得很滑稽：

0、冠軍将赢得一份“神秘大獎”（比如很巨大的一本學生研究論文集……）。

1、排名為素數的學生将赢得最好的獎品 —— 小黃人玩偶！

2、其他人将得到巧克力。

給定比賽的最終排名以及一系列參賽者的 ID，你要給出這些參賽者應該獲得的獎品。

輸入格式：

輸入第一行給出一個正整數 N （ ≤ 1 0 4 ） N（≤10^4） N（≤104），是參賽者人數。随後 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?
           

思路分析：

可以用set中find函數來查找，判斷是否出現

if(set_.find(temp) != set_.end())

是否出現過也可以建立一個數組把出現過的指派為1；

set裡的find函數的傳回值及其用法

代碼：

#include <bits/stdc++.h>
using namespace std;
vector<int> array_(10001), set_(10001);
bool is_prime(int n){
    for(int i = 2; i * i <= n; i++)
        if(n % i == 0)
            return false;
    return true;
}
int main(){
    int n, temp, k;
    cin >> n;
    for(int i = 1; i <= n; i++){
        cin >> temp;
        array_[temp] = i;
    }
    cin >> k;
    for(int i = 0; i < k; i++){
        cin >> temp;
        printf("%04d: ", temp);
        if(array_[temp] == 0){
            cout << "Are you kidding?" << endl;
            continue;
        }
        if(set_[temp] == 0){
            set_[temp] = 1;
        }else{
            cout << "Checked" << endl;
            continue;
        }
        if(array_[temp] == 1){
            cout << "Mystery Award" << endl;
        }else if(is_prime(array_[temp])){
            cout << "Minion" << endl;
        }else{
            cout << "Chocolate" << endl;
        }
    }
    return 0;
}
           

PAT_BasicLevel