C 语言竞赛是浙江大学计算机学院主持的一个欢乐的竞赛。既然竞赛主旨是为了好玩,颁奖规则也就制定得很滑稽:
- 0、冠军将赢得一份“神秘大奖”(比如很巨大的一本学生研究论文集……)。
- 1、排名为素数的学生将赢得最好的奖品 —— 小黄人玩偶!
- 2、其他人将得到巧克力。
给定比赛的最终排名以及一系列参赛者的 ID,你要给出这些参赛者应该获得的奖品。
输入格式:
输入第一行给出一个正整数 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?
思路:
这道题可以利用数组来进行O(1)级别的查找,但是我因为刚接触到 set 函数,因此在题目中尽量是用了 set 和 map。超时是不存在的。这种类型的题目也充分说明了:一个恰当的数据结构可以起到事半功倍的效果。
注意一点,在判断质数的时候,是 i<= sqrt(num) , 切记不要忘了等于号。
CODE
#include <set>
#include <map>
#include <cmath>
#include <iostream>
using namespace std;
bool is_prime(int num)
{
int n = sqrt(num);
for (int i = 2; i <= n; i++)
{
if (num%i==0)
return false;
}
return true;
}
int main()
{
set <string> ss;
map <string,int> ff;
int N, K;
cin >> N;
string ID;
for (int i = 1; i <= N; i++)
{
cin >>ID;
ff[ID] = i;
}
cin >> K;
for (int i = 0; i < K; i++)
{
cin >> ID;
if (!ff.count(ID)) //如果没有,那么输出玩笑
{
cout << ID << ": Are you kidding?" << endl;
}
else
{
if (!ss.count(ID)) //如果不曾输出过,那么准备输出。
{
ss.insert(ID);
int p = ff[ID];
if (p==1) cout << ID << ": Mystery Award" << endl;
else if (is_prime(p)) cout << ID << ": Minion" << endl;
else cout << ID << ": Chocolate" << endl;
}
else
{
cout << ID << ": Checked" << endl;
}
}
}
}