leetcode刷题python之合并K个排序链表

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def mergeKLists(self, lists: List[ListNode]) -> ListNode:
        if lists==[]:return None
        result=ListNode(0)
        prehead=result
        length=len(lists)
        while lists:
            temp=[]
            for i in lists:
                if i!=None:  
                    temp.append(i.val)
                else:
                    temp.append(10000)
            #print(temp)
            if temp!=[10000]*length:
                temp_index=temp.index(min(temp))
                prehead.next=lists[temp_index]
            
                lists[temp_index]=lists[temp_index].next
            
                prehead=prehead.next
                #print(prehead.val)
            else:
                break
        
        return result.next
        
        #prehead.next=lists.index(min(c))
        
        

首先是经过了自己的不懈努力，写出了一个第一个用例没有问题，但是总的来说超时了的程序，应该再多试试，比如迭代，回溯，没准可以更省时间一些

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
class Solution:    
    def mergeKLists(self, lists: List[ListNode]) -> ListNode:
        import heapq
        head=ListNode(0)
        result=head
        stack=[]
        for i in range(len(lists)):
            if lists[i]:
                heapq.heappush(stack,(lists[i].val,i))
                lists[i]=lists[i].next
        while stack:
            #当某物不为空的时候的写法
            val,idx=heapq.heappop(stack)
            result.next=ListNode(val)
            result=result.next
            if lists[idx]:
                heapq.heappush(stack,(lists[idx].val,idx))
                lists[idx]=lists[idx].next
        return head.next

这一次学习了堆heapq的数据结构，这个结构可以给数据将最小的数放在堆顶，通过heappush和heappop进行压入和弹出，这个数据可以是（val，idx）形式，前面是值的大小，idx不参与比较可以作为参数，
可以使用while lists[i]:来解决链表为空的情况
put()将元素添加到序列尾端，get()从队列尾部移除元素。