LeetCode 23 Merge k Sorted Lists（优先队列|链表归并）

题意：归并k个有序链表。

思路：用优先队列存储每个链表的第一个元素，每次将最大的结点放到当前链表的尾部。

/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
struct cmp {
bool operator () (const ListNode* A, const ListNode* B) const {
return A->val > B->val;
}
};
ListNode* mergeKLists(vector<ListNode*>& lists) {
priority_queue<ListNode*, vector<ListNode*>, cmp> q;
ListNode *head = new ListNode(0);
ListNode *tail = head;
for (int i = 0; i < lists.size(); i++) {
if (lists[i] != NULL) {
q.push(lists[i]);
}
}
while (!q.empty()) {
tail->next = q.top();
q.pop();
tail = tail->next;
if (tail->next != NULL)
q.push(tail->next);
}
return head->next;
}
};