目录
23. 合并 K 个升序链表
题目描述:
实现代码与解析:
优先级队列:
原理思路:
23. 合并 K 个升序链表
题目描述:
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 1:
输入:lists = [[1,4,5],[1,3,4],[2,6]] 输出:[1,1,2,3,4,4,5,6] 解释:链表数组如下: [1->4->5,1->3->4,2->6 ] 将它们合并到一个有序链表中得到。 1->1->2->3->4->4->5->6
示例 2:
输入:lists = [] 输出:[]
示例 3:
输入:lists = [[]] 输出:[]
提示:
k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i]
按 升序 排列lists[i].length
的总和不超过10^4
实现代码与解析:
优先级队列:
/*** Definition for singly-linked list.* struct ListNode {* int val;* ListNode *next;* ListNode() : val(0), next(nullptr) {}* ListNode(int x) : val(x), next(nullptr) {}* ListNode(int x, ListNode *next) : val(x), next(next) {}* };*/
class Solution {
public:struct Node{int val;ListNode* ptr;bool operator < (const Node &node) const{return val > node.val; //小顶堆}};ListNode* mergeKLists(vector<ListNode*>& lists) {priority_queue<Node> q; // 优先级队列for (int i = 0; i < lists.size(); i++){if (lists[i]) q.push({lists[i]->val, lists[i]}); // 入队}ListNode* head = new ListNode(); // 头节点ListNode* cur = head;while(q.size()){auto t = q.top(); q.pop(); // 出队cur->next = t.ptr;cur = cur->next;auto nt = t.ptr->next; if (nt) q.push({nt->val, nt}); // 已经出队的节点将其下一个节点入队}return head->next;}
};
原理思路:
优先级队列,小顶堆,定义一个结构体,里面存有节点值用于堆的比较,指针,用于记录链表中节点的位置,每次取出节点,记得把其后面相连的节点入队比较,直到为空为止。很简单,不再详细解释了。