K路归并排序。注意各种边界:比如指针为NULL或者vector空。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
class Solution {
public:
ListNode *mergeKLists(vector<ListNode *> &lists) {
if(lists.size() <= 0)
return NULL;
while(lists.size() > 1) {
vector<ListNode *>lists_tmp;
lists_tmp.clear();
ListNode * tmp;
int i = 0;
while (true) {
tmp = merge2Lists(lists[2*i],lists[2*i+1]);
lists_tmp.push_back(tmp);
i++;
if(2*i+1 == lists.size()) {
lists_tmp.push_back(lists[2*i]);
break;
}
if(2*i+1 > lists.size()) {
break;
}
}
lists.clear();
lists = lists_tmp;
}
return lists[0];
}
private:
ListNode *merge2Lists(ListNode * list1,ListNode * list2) {
ListNode * list = NULL;
ListNode * list_c = NULL;
while(list1 != NULL && list2 != NULL) {
if(list1->val < list2->val) {
if(list == NULL) {
list = list1;
list_c = list;
}
else {
list_c->next = list1;
list_c = list_c->next;
}
list1 = list1->next;
}
else {
if(list == NULL) {
list = list2;
list_c = list;
}
else {
list_c->next = list2;
list_c = list_c->next;
}
list2 = list2->next;
}
}
while(list1!=NULL) {
if(list_c == NULL) {
list_c = list1;
list = list_c;
list1 = list1->next;
continue;
}
list_c->next = list1;
list_c= list_c->next;
list1 = list1->next;
}
while(list2!=NULL) {
if(list_c == NULL) {
list_c = list2;
list = list_c;
list2 = list2->next;
continue;
}
list_c->next = list2;
list_c= list_c->next;
list2 = list2->next;
}
if(list_c != NULL)list_c->next = NULL;
return list;
}
};
|