PAT乙级_1059 C语言竞赛 (20 分)_python

题目：

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?

思路：因为题目中有提到素数这个条件，所以定义了 isPrame() 函数来判断是否为素数，其次用字典来存储ID和对应的名次，最后要注意的就是已查过的ID进行一个处理，本答案中是对它进行一个赋值 -1 的处理，详细解释看代码。

def isPrime(num):
    # 先排除不是2的偶数
    if num != 2 and num % 2 == 0:
        return False
    import math
    flag = True
    for j in range(3, int(math.sqrt(num)) + 1, 2):
        if num % j == 0:
            flag = False
            break
    return flag


N = int(input())
# 用一个字典存储ID和对应的名次
dic = {}
for i in range(N):
    ID = input()
    dic[ID] = i + 1

K = int(input())
lst = []
for i in range(K):
    query = input()
    if query in dic.keys():
        rank = dic[query]
        if rank == 1:
            msg = 'Mystery Award'
        elif rank == -1:
            msg = 'Checked'
        elif isPrime(rank):
            msg = 'Minion'
        else:
            msg = 'Chocolate'
        # ID已查过赋值-1
        dic[query] = -1
    else:
        msg = 'Are you kidding?'
    print('%04s: %s' % (query, msg))