Merge k Sorted Lists

最新推荐文章于 2019-10-31 10:48:55 发布

原创最新推荐文章于 2019-10-31 10:48:55 发布 · 581 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#优先队列 #归并排序

Leetcode 专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一种使用优先队列合并K个已排序链表的方法，并分析了其时间复杂度为n*log(k)，其中n为所有元素总数，k为链表数量。通过此方法可以有效地将多个排序链表合并成一个单一的排序链表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

Tag：使用优先队列，时间复杂度为n*log(k)，其中n为所有元素个数，k为已排好序的链表，如果k很小的话，实现复杂度近乎为线性。

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

struct ListNode
{
	int val;
	ListNode *next;
	ListNode(int x) : val(x), next(NULL) {}
};

struct cmp
{
	bool operator () (ListNode * const &p1, ListNode * const &p2)
	{
		return p1->val > p2->val;
	}
};

ListNode * mergeKLists(vector<ListNode *> &lists)
{
    if(lists.size() == 0)
        return NULL;

	priority_queue<ListNode *, vector<ListNode *>, cmp> heap;
	ListNode * head = NULL;
	ListNode * p = NULL;

	for(int i = 0; i < lists.size(); i++)
	{
	    if(lists != NULL)
            heap.push(lists[i]);
	}

	while(!heap.empty())
	{
		if(head == NULL)
		{
			head = heap.top();
			p = head;
		}
		else
		{
			p->next = heap.top();
			p = p->next;
		}

		if(heap.top()->next != NULL)
		{
			heap.push(heap.top()->next);
		}
		heap.pop();
	}

	return head;
}