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

最新推荐文章于 2025-02-18 20:04:54 发布

leileii

最新推荐文章于 2025-02-18 20:04:54 发布

阅读量716

点赞数

分类专栏： leetcode_python

本文链接：https://blog.youkuaiyun.com/leileii/article/details/93753810

版权

leetcode_python 专栏收录该内容

106 篇文章

订阅专栏

本文介绍了使用Python解决LeetCode中合并K个排序链表的问题。作者分享了自己尝试解题的过程，提到初始解决方案虽然能解决部分用例但存在超时问题。学习并应用了heapq数据结构，了解到该结构如何保持堆顶为最小元素，以及heappush和heappop的操作。文章还提到了处理链表为空的策略以及put和get在队列操作中的作用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 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()从队列尾部移除元素。