给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 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
示例 2:
输入:lists = []
输出:[]
示例 3:
输入:lists = [[]]
输出:[]
提示:
k == lists.length0 <= k <= 10^40 <= lists[i].length <= 500-10^4 <= lists[i][j] <= 10^4lists[i]按 升序 排列lists[i].length的总和不超过10^4
思路
先合并两个升序链表为一个升序链表,然后再用合并后的链表合并第三个升序链表,依次类推。
总的算法是:合并两个有序链表
cpp
class Solution {
public:
//合并两个升序的链表
ListNode* merge(ListNode* head1, ListNode* head2) {
if(!head1) return head2;
if(!head2) return head1;
ListNode prehead(0);
ListNode* pMove=&prehead;
while(head2&&head1){
if(head1->val<=head2->val){
pMove->next=head1;
head1=head1->next;
}
else{
pMove->next=head2;
head2=head2->next;
}
pMove=pMove->next;
}
if(head2)
pMove->next=head2;
if(head1)
pMove->next=head1;
return (&prehead)->next;
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
int size=lists.size();
if(0==size) return nullptr;
if(1==size) return lists[0];
ListNode* list=nullptr;
for(int i=0;i<size;i++){
list=merge(list,lists[i]);
}
return list;
}
};
推荐一个零声教育学习教程,个人觉得老师讲得不错,分享给大家:[Linux,Nginx,ZeroMQ,MySQL,Redis,fastdfs,MongoDB,ZK,流媒体,CDN,P2P,K8S,Docker,TCP/IP,协程,DPDK等技术内容,点击立即学习:链接