PAT1 1012 The Best Rank

题目链接 我的github

题目大意

给出学生的C、M、E的成绩，要求出其中某些学生的最好排名（C、M、E、A，A是C、M、E的平均值，最好排名是指在C、M、E、A这四个排名中最小的那个）

输入

每组包含一个测试用例，每个用例的第一行是2个数$N$和$M$($\le 2000$)，分别表示学生的总数和查询最好排名的学生数。接着$N$行，每一行是学生的ID（长度为6的字符串），然后跟着3个整数（范围是$[0,100]$）表示学生的C、M、E成绩。之后又有$M$行，每行是一个学生的ID

输出

对要查询的$M$个学生在一行中输出他的最好排名和最好排名是哪一门（C、M、E、A），如果最好排名不止一门就按照A>C>M>E的优先级输出最高的那一门，如果要查询的学生ID不在给的$N$个学生之中，就输出N/A

样例输入

5 6 310101 98 85 88 310102 70 95 88 310103 82 87 94 310104 91 91 91 310105 85 90 90 310101 310102 310103 310104 310105 999999

样例输出

1 C 1 M 1 E 1 A 3 A N/A

解析

按照优先级排序就完事了，注意A的成绩是其他三门成绩的平均值并四舍五入

class student: def __init__(self, C, M, E, ID): self.grade = [round((C + M + E) / 3), C, M, E] self.id = ID self.rank = [-1, -1, -1, -1] def solve(): line = input().split(" ") n = int(line[0]) m = int(line[1]) students = {} for i in range(n): line = input().split(" ") ID = line[0] C = int(line[1]) M = int(line[2]) E = int(line[3]) s = student(C, M, E, ID) students[ID] = s for x in range(4): p = sorted(students.values(), key=lambda i: -i.grade[x]) p[0].rank[x] = 1 for i in range(1, n): p[i].rank[x] = i + 1 if p[i].grade[x] == p[i - 1].grade[x]: p[i].rank[x] = p[i - 1].rank[x] for i in range(m): line = input() try: unit = students[line] temp = zip(unit.rank, ['0A', '1C', '2M', '3E']) temp = sorted(temp) print(str(min(unit.rank)), temp[0][1][1]) except: print('N/A') if __name__ == "__main__": solve()