【LeetCode】23. Merge k Sorted Lists（C++）

最新推荐文章于 2024-04-03 21:24:25 发布

原创最新推荐文章于 2024-04-03 21:24:25 发布 · 592 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

LeetCode 专栏收录该内容

82 篇文章

订阅专栏

本文介绍了一种使用小根堆实现的算法，用于合并k个有序链表为一个有序链表。通过将链表头结点加入堆中，不断弹出最小元素并将其下一个结点加入堆，直至所有结点被处理。

地址：https://leetcode.com/problems/merge-k-sorted-lists/

题目：

Merge $k$ sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

Example:
在这里插入图片描述

理解：

合并k个有序链表。想到了外部排序里面的败者数，其本质也是一个堆。因此，把k个链表的头结点加到一个小根堆里，然后把堆顶弹出，并把堆顶的下一个结点加入。

实现：

class Solution {
public:
	ListNode* mergeKLists(vector<ListNode*>& lists) {
		auto comp = [](const ListNode* lhs, const ListNode* rhs) {return lhs->val > rhs->val; };
		priority_queue<ListNode*, vector<ListNode*>, decltype(comp)> q(comp);
		for (auto p : lists)
			if (p)
				q.push(p);
		if (q.empty()) return nullptr;

		ListNode* head = q.top();
		q.pop();
		if (head->next)
			q.push(head->next);
		ListNode* p = head;
		while (!q.empty()) {
			p->next = q.top();
			q.pop();
			p = p->next;
			if (p->next) q.push(p->next);
		}
		return head;
	}
};