23.合并K个排序链表

最新推荐文章于 2020-06-30 22:25:26 发布

一纸一清风

最新推荐文章于 2020-06-30 22:25:26 发布

阅读量109

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/Luckyboom_/article/details/101447442

版权

LeetCode 专栏收录该内容

50 篇文章 0 订阅

订阅专栏

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
	class CMP
	{
	public:
		bool operator()(const ListNode* node1,const ListNode* node2)
		{
			return node1->val < node2->val;
		}
	};

	ListNode* mergeKLists(vector<ListNode*>& lists) {
		ListNode* p;
		vector<ListNode*> v;
		for (int i = 0; i < lists.size(); ++i)
		{
			p = lists[i];
			while (p){
				v.push_back(p);
				p = p->next;
			}
		}
		if (v.size() == 0)
			return nullptr;
		sort(v.begin(), v.end(), CMP());
		for (int i = 1; i < v.size(); ++i)
		{
			v[i - 1]->next = v[i];
		}
		v[v.size() - 1]->next = nullptr;
		return v[0];
	}
};