leetcode23:合并K个排序链表

题目:

合并 k 个排序链表,返回合并后的排序链表

示例

输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6

思路
使用python的优先队列PriorityQueue实现。python的优先队列是用堆实现的,本质是一个二叉树,进入的元素会自动进行排序。
代码

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None
import Queue
class Solution(object):
    def mergeKLists(self, lists):
        """
        :type lists: List[ListNode]
        :rtype: ListNode
        """
        q = Queue.PriorityQueue()
        head = node = ListNode(0)
        if not lists:
            return 
        for l in lists:
            if l:
                q.put((l.val, l))
        # 这个循环很巧妙,保证了找出k个有序链表中最小的数并将该数所在的链表后移。
        # 队列里面存的始终是k个有序列表的头。
        while not q.empty():
            val, temp = q.get()
            node.next = ListNode(val)
            node = node.next
            temp = temp.next
            if temp:
                q.put((temp.val, temp))
        return head.next
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值