合并K个排序链表

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        int k = lists.size();

        struct cmp
        {
            bool operator()(ListNode* a, ListNode* b)
            {
                return a->val > b->val;
            }
        };

        priority_queue<ListNode *, vector<ListNode*>, cmp> q; // 构建小顶堆

        for (int i = 0; i < k; i++)
        {
            if (lists[i])
                q.push(lists[i]);  // 链表非空，加入队列
        }

        ListNode *head = new ListNode(0); // 新建哨兵结点，方便操作
        ListNode *temp = head;

        while(!q.empty())
        {
            temp->next = q.top(); // 将元素值最小的首结点插入到新链表后面
            temp = temp->next; // 指针后移
            q.pop();        // 首结点元素值最小的链表出队
            if (temp->next)  // 把删除首结点后的链表放入队列中
                q.push(temp->next);
        }

        temp = head;
        head = head->next;
        delete(temp);  // 删除哨兵结点

        return head;
    }
};

采用优先级队列的方式代码。