L1-005 考试座位号 (15分)
每个 PAT 考生在参加考试时都会被分配两个座位号,一个是试机座位,一个是考试座位。正常情况下,考生在入场时先得到试机座位号码,入座进入试机状态后,系统会显示该考生的考试座位号码,考试时考生需要换到考试座位就座。但有些考生迟到了,试机已经结束,他们只能拿着领到的试机座位号码求助于你,从后台查出他们的考试座位号码。
输入格式:
输入第一行给出一个正整数 N(≤1000),随后 N 行,每行给出一个考生的信息:准考证号 试机座位号 考试座位号。其中准考证号由 16 位数字组成,座位从 1 到 N 编号。输入保证每个人的准考证号都不同,并且任何时候都不会把两个人分配到同一个座位上。
考生信息之后,给出一个正整数 M(≤N),随后一行中给出 M 个待查询的试机座位号码,以空格分隔。
输出格式:
对应每个需要查询的试机座位号码,在一行中输出对应考生的准考证号和考试座位号码,中间用 1 个空格分隔。
输入样例:
4
3310120150912233 2 4
3310120150912119 4 1
3310120150912126 1 3
3310120150912002 3 2
2
3 4
输出样例:
3310120150912002 2
3310120150912119 1
代码
N = int(input()) #输入得到正整数N
student = [] #建立一个考生信息列表
for i in range(N) : #循环考生个数
student.append(list(input().split(' '))) #以空格分隔,得到考生的“准考证号 试机座位号 考试座位号”,并将每个考生的信息形成一个单独的列表,添加到student列表中
M = int(input()) #输入得到正整数M
testSeatNumber = []
testSeatNumber.extend(list(input().split(' '))) #随后一行中给出 M 个待查询的试机座位号码,以空格分隔
for j in testSeatNumber : #j依次取需要查询的试机号码
for i in student : #i依次取每个学生的信息
if j == i[1] : #将需要查询的试机号码与每个学生的信息中的试机号码对比,若一样输出结果
print("{} {}".format(i[0], i[2])) #输出对应考生的准考证号i[0]和考试座位号码i[2]
student.remove(i) #移除已经查询的考生信息,…这个操作本来是没有的,后来发现,很容易超时,所以加的。但是,我发现耗时还是有点长…,160ms,我也不知道怎么改了…,欢迎大佬批评指正。