给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 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
题目链接:
https://leetcode-cn.com/problems/merge-k-sorted-lists/
解法:两两合并
思路:这道题的基础是合并两个升序链表,只要会合并两个升序链表,合并 k 个也差不多。
声明一个头结点,然后对链表数组进行遍历,把每次合并结果的头结点,跟下一个链表进行合并,最后得到合并k个链接的结果
代码实现:
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
ListNode ans = null;
for (int i = 0 ; i < lists.length; i++) {
ans = mergetTwoLists(ans, lists[i]);
}
return ans;
}
private ListNode mergetTwoLists(ListNode a, ListNode b) {
if (a == null || b == null) {
return a != null ? a : b;
}
ListNode dummyHead = new ListNode(0);
ListNode tail = dummyHead;
ListNode aPtr = a;
ListNode bPtr = b;
while (aPtr != null && bPtr != null) {
if (aPtr.val < bPtr.val) {
tail.next = aPtr;
aPtr = aPtr.next;
} else {
tail.next = bPtr;
bPtr = bPtr.next;
}
tail = tail.next;
}
// 把剩余的节点拼在后面
tail.next = aPtr != null ? aPtr : bPtr;
return dummyHead.next;
}
}
提交通过了,但是击败率有点低啊,用Go语言试试
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func mergeKLists(lists []*ListNode) *ListNode {
var ans *ListNode
for i := 0; i < len(lists); i++ {
ans = mergeTwoLists(ans , lists[i]);
}
return ans
}
func mergeTwoLists(a *ListNode, b *ListNode ) *ListNode {
// go语言没有三目表达式
if a == nil || b == nil {
if a != nil {
return a
} else {
return b
}
}
dummyHead := &ListNode{
Val: 0,
Next: nil,
}
var tail = dummyHead
var aPtr = a
var bPtr = b
// go语言没有while循环
for {
if aPtr == nil || bPtr == nil {
break
}
if aPtr.Val < bPtr.Val {
tail.Next = aPtr
aPtr = aPtr.Next
} else {
tail.Next = bPtr
bPtr = bPtr.Next
}
tail = tail.Next
}
if aPtr != nil {
tail.Next = aPtr
} else {
tail.Next = bPtr
}
return dummyHead.Next;
}
我使用了 go 语言,但是比不上别人go语言的结果,为什么呢?因为时间复杂度不是最优的,上面解法的复杂度如下:
时间复杂度:O(k^2 * n)
空间复杂度:O(1)
采用分治法或者优先队列可以把 k^2 优化为 k*logk,别人应该是采用了更优的解法
看官方题解就能解了:
https://leetcode-cn.com/problems/merge-k-sorted-lists/solution/he-bing-kge-pai-xu-lian-biao-by-leetcode-solutio-2/
日期:2021-12-04