题目描述:
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
Example:
Input: [ 1->4->5, 1->3->4, 2->6 ] Output: 1->1->2->3->4->4->5->6
代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
int findMini(vector<ListNode*>& lists){
int mini = lists[0] -> val;
for(int i = 1; i < lists.size(); i++){
ListNode* tmp = lists[i];
int num = tmp -> val;
if(num < mini) mini = num;
}
return mini;
}
void initial(vector<ListNode*>& lists){
for(int i = 0; i < lists.size(); i++){
if(lists[i] == NULL){
lists.erase(lists.begin() + i);
i--;
}
}
}
ListNode* mergeKLists(vector<ListNode*>& lists) {
initial(lists);
ListNode* result = new ListNode(0);
ListNode* current = result;
while(lists.size() != 0){
int mini = findMini(lists);
for(int i = 0; i < lists.size(); i++){
while(lists[i] -> val == mini){
ListNode* node = new ListNode(mini);
current -> next = node;
current = current -> next;
lists[i] = lists[i] -> next;
if(lists[i] == NULL){
lists.erase(lists.begin() + i);
i--;
break;
}
}
}
}
return result -> next;
}
};
昨天七夕,浪着浪着就忘记做了(毕竟刚开始0.0),这个算补昨天的吧。本题是一个hard题,AC率有点吓人,才百分之29点多,但是我感觉题目稍微有点问题,在题目开始的结构体定义中,构造函数是必须有一个初始参数int x的,也就是不存在空链表这么一说,但是在我提交了第一次的时候,收到一个WA,原因是当input为[[]]时候,output为[],也就是说本题的输入可以为一个size不为0的,但是链表可以为NULL的vector。要不是这个小bug,博主一次AC了有木有!!!!*★,°*:.☆( ̄▽ ̄)/$:*.°★* 。
好了,废话不多说了,来看解题思路,题目要求合并k个有序链表,并且返回一个有序链表。我的思路,每次找所有链表头结点val的最小值。然后遍历vector,对于vector中的每个链表,进行遍历,如果结点值等于mini,则result链表中添加val为mini的结点,并且将该链表的头结点删除,遍历直到头结点的值不等于mini。或者当该链表为NULL时,将其从vector中erase掉。
findMini函数即每次从所有链表的头结点中找最小值。initial函数就是我开始说的,此题存在空链表,所以通过initial函数将这些空链表直接删除。
静心尽力!