PAT甲级 1028 List Sorting

1028 List Sorting

题目大意：给出一组学生数据，由学号，姓名，分数组成，根据输入的标识符c进行相应的排序

思路：简单模拟。不过我第一次提交最后一个测试点超时，第二次又通过了，应该是耗时边界吧，398ms

def cmp(t,i):
    return (t[i-1],t[0])

n,c=map(int,input().split())
l=[]
for i in range(n):
    tmp=input().split()
    tmp[2]=int(tmp[2])
    l.append(tuple(tmp))
l.sort(key=lambda x:cmp(x,c))
for i in l:
    print(*i)