文章目錄

• • 1. 題目信息
  • 2. 思路
  • 3. 代碼

1. 題目信息

合并 k 個排序鏈表，返回合并后的排序鏈表。請分析和描述算法的復雜度。

示例:輸入: [1->4->5,1->3->4,2->6 ] 輸出: 1->1->2->3->4->4->5->6

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/merge-k-sorted-lists
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 思路

建立優先隊列（小頂堆）

將每個鏈表的隊首指針push進入優先隊列，O(k)

取出堆頂，讀取堆頂的值，插入新的鏈表，O(1)

將堆頂的next指針（如果存在）push進入優先隊列，O(logk)

彈出堆頂，O(logk)

循環以上 3-6 ，復雜度*n(總的結點個數)
總的時間復雜度O(n ? logk)

3. 代碼

/*** Definition for singly-linked list.* struct ListNode {* int val;* ListNode *next;* ListNode(int x) : val(x), next(NULL) {}* };*/ class Solution { public:struct cmp{bool operator()(ListNode *a, ListNode *b){return a->val > b->val;}};ListNode* mergeKLists(vector<ListNode*>& lists){priority_queue<ListNode*, vector<ListNode*>, cmp> queue;ListNode *head = new ListNode(0);ListNode *temp = head, *topNext;for(int i = 0; i < lists.size(); ++i){if(lists[i])queue.push(lists[i]);}while(!queue.empty()){temp->next = queue.top();temp = temp->next;topNext = queue.top()->next;queue.pop();if(topNext)queue.push(topNext);}return head->next;} };

總結

以上是生活随笔為你收集整理的LeetCode 23. 合并K个排序链表（优先队列）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。