其实整体思想和归并k个有序数组是一样的,有三种方法,效率最慢的是两个有序链表和并之后再与其他链表合并,需要合并n-1次。第二种方法是两两一组分别归并,第三种方法是采用一个k大小的堆,刚开始的时候将每个链表的第一个元素都放入k的最小堆中,那么就可以直接得到k中最小的值,再将k中刚取出的最小值的下一个节点加入堆中,直到所有的元素都处理完成。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* p = l1;
ListNode* q = l2;
ListNode* newl = new ListNode(0);
ListNode* temp = newl;
while(p!=NULL && q != NULL)
{
if(p->val<q->val)
{
newl->next = new ListNode(p->val);
newl = newl->next;
p = p->next;
}
else
{
newl->next = new ListNode(q->val);
newl = newl->next;
q = q->next;
}
}
if(p != NULL)
newl->next = p;
if(q != NULL)
newl->next = q;
return temp->next;
}
/*ListNode* mergeKLists(vector<ListNode*>& lists) {
int n = lists.size();
if(n<=0) return NULL;
if(n == 1) return lists[0];
ListNode* first = lists[0];
for(int i=1;i<n;i++)
{
first = mergeTwoLists(first,lists[i]);
}
return first;
}*/
ListNode* mergeKLists(vector<ListNode*>& lists) {
int n = lists.size();
if(n<=0) return NULL;
if(n == 1) return lists[0];
vector<ListNode*>::iterator mid = lists.begin()+n/2;
vector<ListNode*> fhalf(lists.begin(),mid);
cout<<"1:"<<fhalf.size()<<endl;
vector<ListNode*> ahalf(mid,lists.end());
cout<<"2:"<<ahalf.size()<<endl;
ListNode* list1 = mergeKLists(fhalf);
ListNode* list2 = mergeKLists(ahalf);
//return list1;
return mergeTwoLists(list1,list2);
}
};