我正在尝试合并两个排序的链表。我想将根指定为两个头中较大的一个。我如何知道哪个元素是最大值,以便我可以将其递增到下一个节点?
目前我正在使用此代码
root = ListNode(max(l1.val, l2.val))
您正在寻找带有密钥max
:
root = ListNode(max(l1, l2, key=lambda x: x.val).val)
max
函数将返回使用时具有最高值的节点,通过此key
。
或者,您可以使用operator.attrgetter
:
import operator
root = ListNode(max(l1, l2, key=operator.attrgetter('val')).val)