Leetcode-Merge k Sorted Lists

forest小拳拳

于 2017-06-13 17:27:12 发布

阅读量172

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： Leetcode

本文链接：https://blog.youkuaiyun.com/qq_33097439/article/details/73188808

Leetcode 专栏收录该内容

97 篇文章

订阅专栏

本文介绍了一种算法，用于合并多个已排序的链表并返回一个单一的排序链表。该方法首先定义了一个辅助函数mergeTwoLists来合并两个链表，然后使用此函数递归地合并所有链表。复杂度分析指出，对于k个链表，总的时间复杂度为O(n log k)，其中n是所有链表中节点的总数。

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

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

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
	ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
		ListNode *l = NULL, *head = NULL, *p = l1, *q = l2;
		if (l1 == NULL)
			return l2;
		if (l2 == NULL)
			return l1;

		while (p && q)
		{
			if (head == NULL)
			{
				if (p->val < q->val)
				{
					head = l = p;
					p = p->next;
					l->next = NULL;
				}
				else
				{
					head = l = q;
					q = q->next;
					l->next = NULL;
				}
			}
			else
			{
				if (p->val < q->val)
				{
					l->next = p;
					l = p;
					p = p->next;
					l->next = NULL;
				}
				else
				{
					l->next = q;
					l = q;
					q = q->next;
					l->next = NULL;
				}
			}

		}
		if (p)
			l->next = p;
		if (q)
			l->next = q;
		return head;
	}
	ListNode* mergeKLists(vector<ListNode*>& lists) {
		if (lists.size() == 0)
			return nullptr;
		while (lists.size() > 1) {
			lists.push_back(mergeTwoLists(lists[0], lists[1]));
			lists.erase(lists.begin());
			lists.erase(lists.begin());
		}
		return lists.front();
	}
};