C++實現(xiàn)LeetCode(23.合并k個有序鏈表)
[LeetCode] 23. Merge k Sorted Lists 合并k個有序鏈表
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
Example:
Input:
[
1->4->5,
1->3->4,
2->6
]
Output: 1->1->2->3->4->4->5->6
這道題讓我們合并k個有序鏈表,最終合并出來的結果也必須是有序的,之前做過一道 Merge Two Sorted Lists,是混合插入兩個有序鏈表。這道題增加了難度,變成合并k個有序鏈表了,但是不管合并幾個,基本還是要兩兩合并。那么首先考慮的方法是能不能利用之前那道題的解法來解答此題。答案是肯定的,但是需要修改,怎么修改呢,最先想到的就是兩兩合并,就是前兩個先合并,合并好了再跟第三個,然后第四個直到第k個。這樣的思路是對的,但是效率不高,沒法通過 OJ,所以只能換一種思路,這里就需要用到分治法 Divide and Conquer Approach。簡單來說就是不停的對半劃分,比如k個鏈表先劃分為合并兩個 k/2 個鏈表的任務,再不停的往下劃分,直到劃分成只有一個或兩個鏈表的任務,開始合并。舉個例子來說比如合并6個鏈表,那么按照分治法,首先分別合并0和3,1和4,2和5。這樣下一次只需合并3個鏈表,再合并1和3,最后和2合并就可以了。代碼中的k是通過 (n+1)/2 計算的,這里為啥要加1呢,這是為了當n為奇數(shù)的時候,k能始終從后半段開始,比如當 n=5 時,那么此時 k=3,則0和3合并,1和4合并,最中間的2空出來。當n是偶數(shù)的時候,加1也不會有影響,比如當 n=4 時,此時 k=2,那么0和2合并,1和3合并,完美解決問題,參見代碼如下:
解法一:
class Solution { public: ListNode* mergeKLists(vector<ListNode*>& lists) { if (lists.empty()) return NULL; int n = lists.size(); while (n > 1) { int k = (n + 1) / 2; for (int i = 0; i < n / 2; ++i) { lists[i] = mergeTwoLists(lists[i], lists[i + k]); } n = k; } return lists[0]; } ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) { ListNode *dummy = new ListNode(-1), *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; } if (l1) cur->next = l1; if (l2) cur->next = l2; return dummy->next; } };
我們再來看另一種解法,這種解法利用了最小堆這種數(shù)據(jù)結構,首先把k個鏈表的首元素都加入最小堆中,它們會自動排好序。然后每次取出最小的那個元素加入最終結果的鏈表中,然后把取出元素的下一個元素再加入堆中,下次仍從堆中取出最小的元素做相同的操作,以此類推,直到堆中沒有元素了,此時k個鏈表也合并為了一個鏈表,返回首節(jié)點即可,參見代碼如下:
解法二:
class Solution { public: ListNode* mergeKLists(vector<ListNode*>& lists) { auto cmp = [](ListNode*& a, ListNode*& b) { return a->val > b->val; }; priority_queue<ListNode*, vector<ListNode*>, decltype(cmp) > q(cmp); for (auto node : lists) { if (node) q.push(node); } ListNode *dummy = new ListNode(-1), *cur = dummy; while (!q.empty()) { auto t = q.top(); q.pop(); cur->next = t; cur = cur->next; if (cur->next) q.push(cur->next); } return dummy->next; } };
下面這種解法利用到了混合排序的思想,也屬于分治法的一種,做法是將原鏈表分成兩段,然后對每段調(diào)用遞歸函數(shù),suppose 返回的 left 和 right 已經(jīng)合并好了,然后再對 left 和 right 進行合并,合并的方法就使用之前那道 Merge Two Sorted Lists 中的任意一個解法即可,這里使用了遞歸的寫法,而本題解法一中用的是迭代的寫法,參見代碼如下:
解法三:
class Solution { public: ListNode* mergeKLists(vector<ListNode*>& lists) { return helper(lists, 0, (int)lists.size() - 1); } ListNode* helper(vector<ListNode*>& lists, int start, int end) { if (start > end) return NULL; if (start == end) return lists[start]; int mid = start + (end - start) / 2; ListNode *left = helper(lists, start, mid); ListNode *right = helper(lists, mid + 1, end); return mergeTwoLists(left, right); } ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) { if (!l1) return l2; if (!l2) return l1; if (l1->val < l2->val) { l1->next = mergeTwoLists(l1->next, l2); return l1; } else { l2->next = mergeTwoLists(l1, l2->next); return l2; } } };
下面這種解法利用到了計數(shù)排序的思想,思路是將所有的結點值出現(xiàn)的最大值和最小值都記錄下來,然后記錄每個結點值出現(xiàn)的次數(shù),這樣從最小值遍歷到最大值的時候,就會按順序經(jīng)過所有的結點值,根據(jù)其出現(xiàn)的次數(shù),建立相對應個數(shù)的結點。但是這種解法有個特別需要注意的地方,那就是合并后的鏈表結點都是重新建立的,若在某些情況下,不能新建結點,而只能交換或者重新鏈接結點的話,那么此解法就不能使用,但好在本題并沒有這種限制,可以完美過 OJ,參見代碼如下:
解法四:
class Solution { public: ListNode* mergeKLists(vector<ListNode*>& lists) { ListNode *dummy = new ListNode(-1), *cur = dummy; unordered_map<int, int> m; int mx = INT_MIN, mn = INT_MAX; for (auto node : lists) { ListNode *t = node; while (t) { mx = max(mx, t->val); mn = min(mn, t->val); ++m[t->val]; t = t->next; } } for (int i = mn; i <= mx; ++i) { if (!m.count(i)) continue; for (int j = 0; j < m[i]; ++j) { cur->next = new ListNode(i); cur = cur->next; } } return dummy->next; } };
到此這篇關于C++實現(xiàn)LeetCode(23.合并k個有序鏈表)的文章就介紹到這了,更多相關C++實現(xiàn)合并k個有序鏈表內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
C++實現(xiàn)讀入二進制數(shù)并轉(zhuǎn)換為十進制輸出
本文給大家介紹的是一則使用C++實現(xiàn)讀入二進制數(shù)并轉(zhuǎn)換為十進制輸出的代碼,實現(xiàn)起來其實非常簡單,C++本身就提供了二進制類庫的,大家看代碼吧,簡單又實用。2015-03-03如何使用arm-none-eabi-gcc編譯器搭建STM32的Vscode開發(fā)環(huán)境
這篇文章主要介紹了使用arm-none-eabi-gcc編譯器搭建STM32的Vscode開發(fā)環(huán)境,本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2023-07-07C++如何將一個vector內(nèi)容賦值給另一個vector,及swap與assign區(qū)別
在本文中,我們將主要介紹5種將一個vector內(nèi)容賦值給另一個vector的方式,順便討論下swap與assign的區(qū)別,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08