力扣 hot100 Day40
23. 合并 K 个升序链表
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
//自己写的垃圾
class Solution {
public:ListNode* mergeKLists(vector<ListNode*>& lists) {vector<int> record;int n = lists.size();for(int i=0;i<n;i++){while(lists[i]){record.push_back(lists[i]->val);lists[i]=lists[i]->next;}}if (record.empty()) {return nullptr;}sort(record.begin(),record.end());ListNode* res = new ListNode(record[0]);ListNode* cur = res;int len = record.size();for(int i=1;i<len;i++){cur->next = new ListNode(record[i]);cur = cur->next;}return res;}
};
没有思考纯粹取巧,放数组里排序后生成新的链表,回去等通知版
//抄的
class Solution {
public:ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {ListNode dummy(0);ListNode* tail = &dummy;while (l1 && l2) {if (l1->val < l2->val) {tail->next = l1;l1 = l1->next;} else {tail->next = l2;l2 = l2->next;}tail = tail->next;}tail->next = l1 ? l1 : l2;return dummy.next;}ListNode* mergeKLists(vector<ListNode*>& lists) {if (lists.empty()) return nullptr;int k = lists.size();while (k > 1) {for (int i = 0; i < k / 2; ++i) {lists[i] = mergeTwoLists(lists[i], lists[k - 1 - i]);}k = (k + 1) / 2;}return lists[0];}
};
面试该写的算法,分治归并算法
逻辑说起来也很简单,两两合并的意思,为了方便循环,一头一尾开始合并,然后逐次减半k值。
比较需要注意的就是k减半的计算,举例子算算就好了
每层的时间复杂度都是 O(N),共有 log₂K 层。总时间复杂度 = O(N log K)。