https://leetcode.cn/problems/merge-k-sorted-lists/
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 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
class Solution:
def mergeKLists(self, lists: List[Optional[ListNode]]) -> Optional[ListNode]:
# 先将多个链表数组合并变成1个数组,排序,再将数组变成链表
nums = []
for lst in lists:
p = lst
while p:
nums.append(p.val)
p = p.next
nums.sort()
# 数组变成链表:倒序版
# pre = None
# for i in range(len(nums) - 1, -1, -1):
# curnode = ListNode(nums[i])
# curnode.next = pre
# pre = curnode
# return pre
# 数组变成链表:正序版
dummy = ListNode()
p = dummy
for i in range(len(nums)):
p.next = ListNode(nums[i])
p = p.next
return dummy.next