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?

Python版本 代码如下

版本1：这份代码运行超时了，原因是我为了统计是否查验过，把查询过的编号追加到列表内了.

import math

n = eval(input())
L={}
for i in range(n):
    L[input()] = i+1

def sushu(m):
    if m!=2 and m%2==0:
        return False
    for i in range(3, int(math.sqrt(m))+1, 2):
        if m%i==0:
            return False
    return True

C=[]
k = eval(input())
for i in range(k):
    search = input()
    if search in L.keys():
    	if search not in C:  ##测试点3：运行超时
	        if L[search]==1:
	            print(search+": Mystery Award")
	        elif sushu(L[search]):
	            print(search+": Minion")
	        else:
	            print(search+": Chocolate")
	        C.append(search)
	    else:
	    	print(search+": Checked")
    else:
        print(search+": Are you kidding?")

版本2：参考了其他人的代码，发现是自己想的麻烦了.

import math

n = eval(input())
L={}
for i in range(n):
    L[input()] = i+1

def sushu(m):
    if m!=2 and m%2==0:
        return False
    for i in range(3, int(math.sqrt(m))+1, 2):
        if m%i==0:
            return False
    return True

k = eval(input())
for i in range(k):
    search = input()
    if search in L.keys():
        if L[search]==1:
            print(search+": Mystery Award")
        elif L[search]==-1:   ##条件2和条件三位置不能换，自行体会
             print(search+": Checked")
        elif sushu(L[search]):
            print(search+": Minion")
        else:
            print(search+": Chocolate")
        L[search]=-1  ##其实简单的赋值-1就可以解决超时问题
    else:
        print(search+": Are you kidding?")