PAT乙级1059.C语言竞赛-python

前言

这个题只是逻辑上复杂点，认真写，不要丢掉某些条件就好了。这道题需要用到素数判断，很耗时间，但是python本来就运行时间较长，最后我的代码有两个超时的，我就不管了，如果大家有更好的方法，欢迎批评。

1. 题目

C 语言竞赛是浙江大学计算机学院主持的一个欢乐的竞赛。既然竞赛主旨是为了好玩，颁奖规则也就制定得很滑稽：

• 冠军将赢得一份“神秘大奖”（比如很巨大的一本学生研究论文集……）。
• 排名为素数的学生将赢得最好的奖品 —— 小黄人玩偶！
• 其他人将得到巧克力。

给定比赛的最终排名以及一系列参赛者的 ID，你要给出这些参赛者应该获得的奖品。

1. 输入格式：
   输入第一行给出一个正整数$N(⩽10^4)$，是参赛者人数。随后$N$行给出最终排名，每行按排名顺序给出一位参赛者的 ID（4 位数字组成）。接下来给出一个正整数 $K$以及$K$个需要查询的 ID。

2. 输出格式：
   对每个要查询的 ID，在一行中输出 ID: 奖品，其中奖品或者是 Mystery Award（神秘大奖）、或者是 Minion（小黄人）、或者是 Chocolate（巧克力）。如果所查 ID 根本不在排名里，打印 Are you kidding?（耍我呢？）。如果该 ID 已经查过了（即奖品已经领过了），打印 ID: Checked（不能多吃多占）。

3. 输入样例：

   6
   1111
   6666
   8888
   1234
   5555
   0001
   6
   8888
   0001
   1111
   2222
   8888
   2222

4. 输出样例：

   8888: Minionbushaoyu
   0001: Chocolate
   1111: Mystery Award
   2222: Are you kidding?
   8888: Checked
   2222: Are you kidding?

2. 代码

有两个超时了，还是求素数的时候超时了估计。

def isPrem(n):
  if n== 2: return True
  for i in range(int(n/2))[2:]:
    if n%i == 0:
      return False
      
  return True
  


N = int(input())
ID = []
check = []

for i in range(N):
  tem = input()
  ID.append(tem)
  
K = int(input())
for i in range(K):
  tem = input()
  if tem not in ID:
    print(tem+": Are you kidding?")
  else:
    index = ID.index(tem)+1
    if index not in check:
      check.append(index)
      if index == 1: print(tem+": Mystery Award")
      elif isPrem(index): print(tem+": Minion")
      else: print(tem+": Chocolate")
    else:
      print(tem+": Checked")
def isPrem(n):
  if n== 2: return True
  for i in range(int(n/2))[2:]:
    if n%i == 0:
      return False
      
  return True
  


N = int(input())
ID = []
check = []

for i in range(N):
  tem = input()
  ID.append(tem)
  
K = int(input())
for i in range(K):
  tem = input()
  if tem not in ID:
    print(tem+": Are you kidding?")
  else:
    index = ID.index(tem)+1
    if index not in check:
      check.append(index)
      if index == 1: print(tem+": Mystery Award")
      elif isPrem(index): print(tem+": Minion")
      else: print(tem+": Chocolate")
    else:
      print(tem+": Checked")