leetcode-腾讯精选练习（50 题）python #23 合并K个排序链表

最新推荐文章于 2024-09-11 02:12:10 发布

楚天星云

最新推荐文章于 2024-09-11 02:12:10 发布

阅读量166

点赞数

分类专栏： leetcode-腾讯精选练习50题

本文链接：https://blog.csdn.net/weixin_43866606/article/details/93711045

版权

leetcode-腾讯精选练习50题专栏收录该内容

49 篇文章 1 订阅

订阅专栏

leetcode-腾讯精选练习（50 题）python #23 合并K个排序链表
题目来源：https://leetcode-cn.com/problemset/50/
合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

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

# 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:
        r, n, p = [], lists and lists.pop(), None
        while lists or n: r[len(r):], n = ([n], n.next or lists and lists.pop()) if n else ([], lists.pop())
        for n in sorted(r, key=lambda x: x.val, reverse=True): n.next, p = p, n
        return n if r else []