先声明一个newhead作为dummy head,最后返回它的next作为答案,然后对于l1, l2线性扫描,每次选两者中值小的那个作为新链表的下一个节点,然后选取的链表前进一步。如果l1, l2里有一个已经结束了,就把另一个剩下的值全部直接连过来。
class Solution:
def mergeKLists(self, lists: List[ListNode]) -> ListNode:
while len(lists)>1:
a = lists.pop() if len(lists)>0 else None
b = lists.pop() if len(lists)>0 else None
lists.insert(0,self.mergeTwoLists(a,b))
return None if len(lists)<1 else lists[0]
def mergeTwoLists(self,l1,l2):
newhead = ListNode(0)
p = newhead
while(l1 and l2):
if l1.val<l2.val:
p.next = ListNode(l1.val)
l1 = l1.next
else:
p.next = listNode(l2.val)
l2 = l2.next
p = p.next
if l1:
p.next = l1
else:
p.next = l2
return newhead.next