1059. C语言竞赛(20)——Python

测试点1与2运行超时

#判断素数
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

#得到获奖名单
num_guys = input()
num_guys = int(num_guys)
guys_list = []
guys_dir = {}
for i in range(num_guys):
guy_id = input()
guys_dir[guy_id] = 0
guys_list.append(guy_id)
guys_dir[guy_id] = i + 1

#判断输入的id获奖情况
judge_guys = input()
judge_guys = int(judge_guys)
for i in range(judge_guys):
id = input()
if id in guys_list and guys_dir[id] > 0:
rank = guys_dir[id]
if rank == 1:
print(id + ': ' + 'Mystery Award')
elif isPrime(rank):

print(id + ': ' + 'Minion')
else:
print(id + ': ' + 'Chocolate')
guys_dir[id] = 0
elif id not in guys_list:
print(id + ': ' + 'Are you kidding?')
else:
print(id + ': ' + 'Checked')