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;
    }
};