题解 | 合并k个已排序的链表
合并k个已排序的链表
https://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6
/**
* struct ListNode {
* int val;
* struct ListNode *next;
* ListNode(int x) : val(x), next(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param lists ListNode类vector
* @return ListNode类
*/
//合并两个有序链表
ListNode* merge(ListNode* a,ListNode* b){
if(a == nullptr)
return b;
if(b == nullptr)
return a;
if(a->val <= b->val){
a->next = merge(a->next, b);
return a;
}
else{
b->next = merge(a, b->next);
return b;
}
}
//分而治之
ListNode* divide(vector<ListNode*>& lists,int left,int right){
//边界条件
if(left == right){
return lists[left];
}
//一分为二,不断往下分
int mid = (left + right)/2;
//递归
return merge(divide(lists, left, mid),divide(lists, mid + 1, right));
}
//归并排序
ListNode* mergeKLists(vector<ListNode*>& lists) {
if(lists.size() == 0)
return nullptr;
return divide(lists, 0, lists.size()-1);
}
};
查看10道真题和解析