题意:把多个有序链表合并成一个有序链表。
分析与思路:题目简明易懂,但是也并非想象中的易事,思维上没有什么复杂的地方,我的思路是直接把第一个和第二个序列合并成一个整体,合并方法是把第二个序列一个个插入第一个链表中。再整体和第三个序列合并,再和第四个,直到结束。每一次合并只需要遍历一次两个链表,所以一次合并复杂度就是O(list1.size+list2.size),若所有链表长度为n,有m个链表,则扫描节点次数:第一次为n+n次,第二次为2n+n,第三次为3n+n,第m-1次为(m-1)n+n=m*n。总扫描节点次数为(2+3+4+5+...+m)*n=(2+m)*(m-1)*n/2
代码:
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
ListNode* merge2Lists(ListNode* list1, ListNode* list2) {//合并两个有序链表
if (list1 == NULL) {//第一个为空则返回第二个序列
list1 = list2;
return list1;
}
ListNode* temp1 = list1;//第一个链表中的指针
ListNode* temp2 = list2;//第二个链表中的指针
ListNode* fa1 = list1;//第一个链表中目前指针的父指针
while (temp1 != NULL&&temp2 != NULL) {
if (temp1->val < temp2->val) {
ListNode* next1 = temp1->next;
if (next1 != NULL&&next1->val >= temp2->val) {//插入到第一个指针的后面
ListNode* newNode = new ListNode(temp2->val);
newNode->next = next1;
temp1->next = newNode;
temp2 = temp2->next;
}
else if (next1 == NULL) {
fa1 = temp1;
temp1 = temp1->next;
continue;
}
else {
fa1 = temp1;
temp1 = temp1->next;
}
}
else {//插入到第一个指针的前面
ListNode* newNode = new ListNode(temp2->val);
newNode->next = temp1;
if (temp1 != list1) {
fa1->next = newNode;
newNode->next = temp1;
}
else {
list1 = newNode;
}
fa1 = newNode;
temp2 = temp2->next;
}
}
if (temp2 != NULL) {//把第二个链表剩下的接入第一个链表后面
fa1->next = temp2;
}
return list1;
}
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
ListNode* re=NULL;
if (lists.size() == 0) return re;
re = lists[0];
int i = 1;
while (i < lists.size()) {
re = merge2Lists(re, lists[i]);
i++;
}
return re;
}
};