B1059 C语言竞赛（python）

1059 C语言竞赛 (20分)
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?

害，日常超时……😳

import math
def isprime(n):
    if n <= 1:
        return False
    for i in range(2, int(math.sqrt(n))+1):
        if n % i == 0:
            return False
    return True
rewards = ['Mystery Award', 'Minion', 'Chocolate']
n = input()
rank = [input() for i in range(int(n))]
k = input()
searchs = [input() for i in range(int(k))]
result = []
checked = []
for search in searchs:
    if search not in rank:# 未找到该ID
        result.append('Are you kidding?')
    elif rank.index(search) == 0 and search not in checked:# 冠军
        result.append(rewards[0])
        checked.append(search)
    elif isprime(rank.index(search)+1) and search not in checked:# 排名为素数
        result.append(rewards[1])
        checked.append(search)
    elif search not in checked:# 巧克力
        result.append(rewards[2])
        checked.append(search)
    else:# 上述情况均不满足，则已经分发过奖品
        result.append('Checked')
for key, value in list(zip(searchs, result)):
    print(key + ': ' + value)