#include <vector>
#include <limits>
// Definition for singly-linked list.
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(nullptr) {}
};
class Solution {
public:
ListNode* mergeKLists(std::vector<ListNode*>& lists) {
if (lists.empty()) return nullptr;
return mergeLists(lists, 0, lists.size() - 1);
}
private:
ListNode* mergeLists(std::vector<ListNode*>& lists, int start, int end) {
if (start == end) return lists[start];
if (start > end) return nullptr;
int mid = start + (end - start) / 2;
ListNode *left = mergeLists(lists, start, mid);
ListNode *right = mergeLists(lists, mid + 1, end);
return mergeTwoLists(left, right);
}
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
if (!l1) return l2;
if (!l2) return l1;
ListNode dummy(-1);
ListNode *cur = &dummy;
while (l1 && l2) {
if (l1->val < l2->val) {
cur->next = l1;
l1 = l1->next;
} else {
cur->next = l2;
l2 = l2->next;
}
cur = cur->next;
}
cur->next = l1 ? l1 : l2;
return dummy.next;
}
};
[leetcode]合并K个升序链表
最新推荐文章于 2024-05-05 09:57:41 发布