我正在尝试合并两个已排序的链接列表。我想将根分配为两个头中较大的一个。我怎么知道哪个元素是最大值所以我可以将它增加到下一个节点?
目前我正在使用此代码
root = ListNode(max(l1.val, l2.val))
答案 0 :(得分:1)
您正在寻找带有密钥的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)