题目描述:
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
使用stl multiset的解法:
/**
* 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:
ListNode* mergeKLists(vector<ListNode*>& lists) {
multiset<int> a;
for(int i = 0;i<lists.size();i++)
{
ListNode *head = lists[i];
while(head!=NULL)
{
a.insert(head->val);
head=head->next;
}
}
ListNode *res = new ListNode(-1);
ListNode *resHead = res;
multiset<int>::iterator it = a.begin();
for(it;it!=a.end();it++)
{
resHead->next = new ListNode(*it);
resHead=resHead->next;
}
return res->next;
}
};