【leetcode】(python) 23. Merge k Sorted Lists

最新推荐文章于 2024-04-03 21:24:25 发布

turbo624

最新推荐文章于 2024-04-03 21:24:25 发布

阅读量126

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/weixin_44110891/article/details/90753113

版权

LeetCode 专栏收录该内容

116 篇文章 0 订阅

订阅专栏

这篇博客讨论了如何使用Python解决LeetCode上的23题——合并k个排序链表。通过堆排序思想，将链表头节点放入小根堆中，每次取出最小值并更新堆，以构建一个有序链表。此外，还提供了另一种解决方案，即先将所有链表的元素放入列表，再对列表排序。这两种方法的时间和内存消耗也进行了比较。

摘要由CSDN通过智能技术生成

Description

Merge k Sorted Lists hard
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

Example

Input:
[
  1->4->5,
  1->3->4,
  2->6
]
Output: 1->1->2->3->4->4->5->6

题目大意

将k个升序链表合成一个升序链表

解体思路

利用堆排序思想，首先将k个链表的头节点放入堆中进行小根堆排序，这样可以得到最小值及其下标号，然后继续将最小数链表的下一个数放入堆中进行小根堆排序，以此类推。

code

# 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:
        head = ListNode(-1)
        move = head
        heap = []
        heapq.heapify(heap)
        [heapq.heappush(heap, (l.val, i)) for i, l in enumerate(lists) if l]
        while heap:
            curVal, curIndex = heapq.heappop(heap)
            curHead = lists[curIndex]
            curNext = curHead.next
            move.next = curHead
            curHead.next = None
            move = curHead
            curHead = curNext
            if curHead:
                lists[curIndex] = curHead
                heapq.heappush(heap, (curHead.val, curIndex))
        return head.next

Runtime：68ms, Memory：16.4 MB

参考较优解

解题思路

依次遍历每个链表并将其值放入一列表中，最后将列表排序即可。

code

buf = []
        for i in range(len(lists)):
            while(lists[i] is not None):
                # print(lists[i].val)
                buf.append(lists[i].val)
                lists[i] = lists[i].next
        buf.sort()
        return buf

Runtime：64ms, Memory：15.8 MB