题目就不描述了,主要讲解思路,题目链接
使用通常思路,开四个列表分别存储各满足要求的数会发现有3个测试点超时,并且需要编写自定义排序函数。
下面给出几点优化:
-
更改存储信息:
通常我们的存储方式是[准考证号,德分,才分],这时候要根据题目要求去自定义排序规则,不过我们只需要改变一下存储的值和顺序,便可以解决这种问题。
现在我们的存储方式是[德分+才分(总分),德分,准考证号],这时候虽然我们少存了一种信息(才分),但是我们可以通过总分 - 德分去动态计算出才分。这时候我们就不需要编写自定义函数去更改排序规则了,因为sort函数对于列表中有多个值是从前往后依次比较排序,这种存储方式刚好满足题目要求。不过还有一点小问题就是前面两个值都是从大到小排序,而准考证号所示从小到大排序,不过我们只需要将准考证号变为负数即可解决问题,排序完成后输出时只需要再添负号变为正数。如a = 100001, a = -a, print(-a) -
更改输入输出流:
由于Python中input()函数有可交互性(括号内可以输出你想在控制台输出的话),它会去掉末尾的换行等,会导致比sys库中的stdin.readline()在百万字符串数据量下慢约10倍。 -
程序整体优化
由于全局变量的开销时间是高于局域变量的,故我们将程序整体放入函数中,这样就可以把全局变量转变为局部变量。 -
更改存储结构(重点)
经过上面三步优化后,我们发现只有2个测试数据点依旧超时。这时候我们或许可以从时间复杂度去优化,通过观察我们发现程序最花费时间的地方就是排序,时间复杂度为O(NlogN),那么能不能给他进一步优化呢?答案是可以的,通过观察我们发现数值比较小,故可以开辟足够大的空间通过下标来存储信息,这样我们得到的信息默认就是从小到大排好序的。
如:“3, 9, 1”,那么我们可以f=[0]*10,f[3]=1,f[9]=1,f[1]=1,之后从小到大遍历,如果值为1那么输出,到这我们就完成了O(N)时间复杂度级别的排序。同理,对于这道题我们发现德分和才分比较小,可以进行这样的优化避免了对前面两个值的排序,由于准考证号数值过大,存储空间不够,我们只需要对这一维数据进行排序即可,实测是可以过的。
-
import sys def p_list(info): # 根据题目要求从大到小输出实际存在的分数 for i in range(201, 0, -1): for j in range(100, 0, -1): # 如果当前有值,则通过sorted排序返回一个列表,用for遍历这个排序好的列表依次输出即可 # 这里存储的是准考证号,要求从小到大 if info[i][j]: for k in sorted(info[i][j]): sys.stdout.write("%s %s %s\n" % (k, j, i - j)) def main(): # 通过列表推导式创建三维列表,为info[][][], 从前往后分别表示总分,德分,准考证号 info1 = [[[] for i in range(101)] for j in range(202)] info2 = [[[] for i in range(101)] for j in range(202)] info3 = [[[] for i in range(101)] for j in range(202)] info4 = [[[] for i in range(101)] for j in range(202)] cnt = 0 n, l, h = map(int, input().split()) a = [list(map(int, sys.stdin.readline().split())) for i in range(n)] for x in a: # 不满足题目要求的学生不予存入 if x[1] < l or x[2] < l: continue # 由于准考证号值过大,无法通过索引形式,只能append添加至末尾 if x[1] >= h and x[2] >= h: info1[x[1] + x[2]][x[1]].append(x[0]) elif x[1] >= h: info2[x[1] + x[2]][x[1]].append(x[0]) elif x[1] >= x[2]: info3[x[1] + x[2]][x[1]].append(x[0]) else: info4[x[1] + x[2]][x[1]].append(x[0]) cnt += 1 sys.stdout.write("%s\n" % cnt) p_list(info1) p_list(info2) p_list(info3) p_list(info4) main()