概述
用大小为k的堆来做,注意堆的使用。
23. 合并K个升序链表
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 1:
输入:lists = [[1,4,5],[1,3,4],[2,6]]
输出:[1,1,2,3,4,4,5,6]
解释:链表数组如下:
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6
code:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def mergeKLists(self, lists):
"""
:type lists: List[ListNode]
:rtype: ListNode
"""
res = []
nhead = ListNode(None)
ret = nhead
for l in lists:
if l:
res.append([l.val, l])
heapq.heapify(res)
while res:
nodeval, node = heapq.heappop(res)
nhead.next = node
nhead = nhead.next
node = node.next
if node:
heapq.heappush(res, [node.val ,node])
return ret.next
最后
以上就是仁爱月光为你收集整理的Leetcode 23. 合并K个升序链表的全部内容,希望文章能够帮你解决Leetcode 23. 合并K个升序链表所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复