/**
* struct ListNode {
* int val;
* struct ListNode *next;
* ListNode(int x) : val(x), next(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param lists ListNode类vector
* @return ListNode类
*/
ListNode* mergeKLists(vector<ListNode*>& lists) {
if (lists.size() == 0) {
return nullptr;
}
ListNode* head = lists[0];
for (int i = 1; i < lists.size(); i++) {
head = mergeList(head, lists[i]);
}
return head;
}
ListNode* mergeList(ListNode* list1, ListNode* list2) {
if (list1 == nullptr || list2 == nullptr) {
return list1 == nullptr ? list2 : list1;
}
ListNode* p, *q;
if (list1->val <= list2->val) {
p = list1;
list1 = list1->next;
} else {
p = list2;
list2 = list2->next;
}
q = p;
while (list1 && list2) {
if (list1->val <= list2->val) {
q->next = list1;
list1 = list1->next;
} else {
q->next = list2;
list2 = list2->next;
}
q=q->next;
}
q->next = list1==nullptr? list2 : list1;
return p;
}
};
07-22
07-22
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交