Python 元组列表排序:初学者可能忽视的细节

P1085 [NOIP2004 普及组] 不高兴的津津

题目描述

津津上初中了。妈妈认为津津应该更加用功学习,所以津津除了上学之外,还要参加妈妈为她报名的各科复习班。另外每周妈妈还会送她去学习朗诵、舞蹈和钢琴。但是津津如果一天上课超过八个小时就会不高兴,而且上得越久就会越不高兴。假设津津不会因为其它事不高兴,并且她的不高兴不会持续到第二天。请你帮忙检查一下津津下周的日程安排,看看下周她会不会不高兴;如果会的话,哪天最不高兴。

输入格式

输入包括 \(7\) 行数据,分别表示周一到周日的日程安排。每行包括两个小于 \(10\) 的非负整数,用空格隔开,分别表示津津在学校上课的时间和妈妈安排她上课的时间。

输出格式

一个数字。如果不会不高兴则输出 \(0\),如果会则输出最不高兴的是周几(用 \(1, 2, 3, 4, 5, 6, 7\) 分别表示周一,周二,周三,周四,周五,周六,周日)。如果有两天或两天以上不高兴的程度相当,则输出时间最靠前的一天。

样例输入

5 3
6 2
7 2
5 3
5 4
0 4
0 6

样例输出

3

解题思路

将数据记录为元组列表,例如:[(1, 2), (3, 4), ...] ,每个元组记录当天的数据,元组第一位为当天的上课时间,第二位为当天的日期(星期几)。

存储数据的方法相同,最后比较输出结果的方法有两种。

方法 1

li = []

for i in range(7):
    a, b = map(int, input().split())
    li.append((a+b, i+1))

li.sort()

if li[6][0] > 8:
    for i in range(6,-1,-1):
        if li[i-1][0] != li[i][0]:
            print(li[i][1])
            break
else:
    print(0)

对列表进行排序,list.sort() 默认升序排列,所以只需要判断最后一个元组内的上课时间是否大于 8 小时,决定是否要输出 0 ;

当上课时间的最大值大于8 时,倒序循环查找目标日期。存在上课时间相同的多个元组时,刚刚的排序操作会将日期较小的放在前面,所以需要在循环内去判断 i-1 位置是否等于 i 位置。

如果不等,输出当前元组的日期并跳出循环即可;如果相等,进入下一轮循环继续比较。

方法 2

li = []

for i in range(7):
    a, b = map(int, input().split())
    li.append((a+b, i+1))

if max(li)[0] > 8:
    print(max(li, key=lambda li:li[0])[1])
else:
    print(0)

该方法使用 max 函数 进行最大值选取。

与刚刚的排序方法不同,max() 只输出 元组内元素之和最大 的元组,也就是说如果存在:[(9, 3), (9, 5)] ,它会返回 (9, 5) 这个元组,我们就无法得到真正的答案 (9,3)。

这时候就需要使用指定选取最大值的方法(函数),通过 key=函数() 来实现,函数部分实现 返回元组内第一个元素 即可,此处使用了匿名函数 lambda 来实现,对应普通函数为:

def takeFirst(li):
    return li[0]

对于 key 的理解,我最初走入了一个误区:li[0] 不是列表 li 的第一个元素吗,应该是一个元组, 这怎么能用来排序啊?

抱着这个疑问我查阅了一些资料,大多说的比较笼统,直到看到这篇文章: 理解sort()函数中的key ,我终于是理解了。

key 关键字指向的函数,会在排序前先执行。换句话说,它会在遍历迭代对象时,对迭代对象中的 每一个元素 进行一次操作。

假如有列表 li = [(1,2), (5,6), (3,4)] ,对其进行排序(sort函数 和 max函数 都可以设置 key ),设置 key为上方的 takeFirst() 函数。

当对 li 进行排序时,首先遍历列表中的元素(元组),对每个元素均执行 takeFirst() 函数,也就是说选出 [1, 5, 3] 来,对其进行排序得到 [1, 3, 5] ,再将这个顺序对应到 li 上,就可以得到排序后的列表 li = [(1,2), (3,4), (5,6)] ,至此排序完成。

写在最后

第一次遇到 key 参数时(刷题笔记 - 1043.[编程入门]三个数字的排序),只是简单的记录了一下,并没有过多深究。但知识就是这样,今天无视它,改天它就掐住了你的脖子。

顺便摘抄一段最近看到的话:

“我非常喜欢的一个东西,是一个人十三四岁的夏天,在路上捡到一支真枪。因为年少无知,天不怕地不怕,他扣下扳机。没有人死,也没有人受伤。他认为自己开了空枪。后来他三十岁或者更老,走在路上听到背后有隐隐约约的风声。他停下来,回过身去,子弹正中眉心。”

“果然,在那明媚的阳光中传来了那一声枪响。那枪声沉闷之极。”

——《我与地坛》史铁生