题目:

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（不能多吃多占）。

输入样例：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
6
1111
6666
8888
1234
5555
0001
6
8888
0001
1111
2222
8888
2222

输出样例：

1
2
3
4
5
6
8888: Minion
0001: Chocolate
1111: Mystery Award
2222: Are you kidding?
8888: Checked
2222: Are you kidding?

代码长度限制：16 KB

时间限制：200 ms

内存限制：64 MB

解题思路：

数据结构

使用哈希映射将每个ID映射对应的奖励 python的defaultdict

查找

查找ID对应的奖励，然后将对应ID的 value 更改为 Checked

若对应ID不存在则返回 Are you kidding?

代码:

Python

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# is primer
def is_primer(num):
    from math import sqrt
    for i in range(2, int(sqrt(num)) + 1):
        if num % i == 0:
            return False
    return True


# input
from collections import defaultdict

N = eval(input())
ID = defaultdict(str)

# 记录每个ID的排名所对应的奖品
for cnt in range(1, N + 1):
    ID_input = input()
    if cnt == 1:
        ID[ID_input] = "Mystery Award"
    elif is_primer(cnt):
        ID[ID_input] = "Minion"
    else:
        ID[ID_input] = "Chocolate"

# search
N = eval(input())
for _ in range(N):
    ID_input = input()
    if ID_input in ID:
        print(f"{ID_input}: {ID[ID_input]}")
        ID[ID_input] = "Checked"
    else:
        print(f"{ID_input}: Are you kidding?")

时间复杂度：O(N)

空间复杂度：O(N)