1059 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?

 看见这种输出格式的第一反应就是用字典结构存储

一开始自己写的代码一个点超时，一个点错误，下面这个大体上是xutiantian1412的代码，我的思路和他的基本上一样，现在想起来好像是我的判断素数的函数没有先排掉偶数，导致了超时，而且没有加上x=1的情况，如果不加的话这个函数在x=1的时候返回是True

import math
N=int(input())
dic={}
for i in range(N):
    name=input()
    dic[name]=i+1

def isPrime(x):     #加上x=1的情况
    if x==1:        #再排除除了2的偶数，否则会超时
        return False
    if x !=2 and x%2==0:#不为2的偶数
        return False
    flag=True
    for i in range(3,int(math.sqrt(x))+1,2):
        if x%i ==0:
            flag=False
            break
    return flag
K=int(input())
for i in range(K):
    check=input()
    if check in dic:
        rank=dic[check]
        if rank==1:
            out='Mystery Award'
        elif rank==-1:
            out='Checked'
        elif isPrime(rank):
            out='Minion'
        else:
            out='Chocolate'
        dic[check]=-1 #每次走完都把key的值置为-1==>核对过的都是-1，下次循环的时候如果碰到-1
    else:                                              #就输出checked
        out='Are you kidding?'
    print('%s: %s' %(check,out))