PTA 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?

解题思路

这题的思路比较简单，其实利用map存放键值对即可。开始时通过键值对存放每一个学生的id和名次（由于输入是按照排名顺序输入的，所以名次就是每轮迭代的轮数，轮数从1开始循环到N）。存放好每个学生的信息后输入K对K个学生进行查询。利用map的find方法可以直接返回键为key的值，在这里进行以下判断，如果值是0说明该同学没有排名，直接输出该同学的id以及对应的输出语句。利用set保存每次查询的获奖学生，因为set的特性是不会保存相同的元素，所以在每轮进行判断当该键即学生id在set中且不是set的最后一个元素（表示列表中有该元素已经领取过奖品）所以输出Checked。其他的检查方法比较简单，通过isPrime函数判断查找同学的名次是否为第一名、素数排名、其他排名对应输出相应的输出。

代码如下

#include<math.h>
#include<iostream>
#include<map>
#include<iomanip>
#include<set>

using namespace std;

bool isPrime(int n) {
    if(n==1)
		return false;
    if(n==2||n==3)
		return true;
    for(int i=2;i<=sqrt(n);i++)
	{
		if(n%i==0)
			return false;
	}
    return true;
}

int main()
{
    int n, t, k;
    map<int, int> mp;
    set<int> st;
    cin>>n;
    for (int i = 1; i <= n; ++i) {
        cin>>t;
        mp.insert(make_pair(t, i));		// 插入键值对，键为学生id，值为排名
    }
    cin>>k;
    for (int i = 1; i <= k; i++) {
        cin>>t;
        cout<<setw(4)<<setfill('0')<<t;		// 控制输出格式，输出4位，用0前面补齐位数
        if (mp[t] == 0) 
		{
            cout<<": Are you kidding?\n";
            continue;
        }
        if (st.find(t) != st.end())	// 判断set集合中是否有该元素
		{
			cout<<": Checked\n";
		}
        else if (mp[t] == 1)
		{
			cout<<": Mystery Award\n";
		}
        else if (isPrime(mp[t]))		// 判断是否为素数
		{
			cout<<": Minion\n";
		}
        else
		{
			cout<<": Chocolate\n";
		}
        st.insert(t);
    }
    return 0;
}

运行结果