比较好的两种思路
1、与合并两个有序链表一样,K个链表一起从头往尾走,每次选取K个链表中指针指向位置的最小值,这个最小值使用优先队列维护较为合适。
2、链表之间两两合并,可复用两个有序链表合并的代码。
从第一个链表开始,依次合并后面的链表
分治,第一个链表和第二个链表合并,第三个、第四个合并,如此下去
基本的还是两个有序链表的合并。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
int size = lists.size();
if(size ==0) return nullptr;
else if(size ==1) return lists[0];
else if(size == 2) return mergesort(lists[0],lists[1]);
else
{
ListNode* res = mergesort(lists[0],lists[1]);
for(int i=2;i<size;i++)
res = mergesort(res,lists[i]);
return res;
}
}
ListNode* mergesort(ListNode*l1,ListNode*l2)
{
ListNode* newlist = new ListNode(0);
ListNode* node = newlist;
while(l1 && l2)
{
if(l1->val<l2->val)
{ node->next = l1;
l1 = l1->next;}
else
{
node->next = l2;
l2 = l2->next;
}
node = node->next;
}
if(l1)
node->next = l1;
if(l2)
node->next = l2;
return newlist->next;
}
};