leetcode23.合并k个有序链表 c++ 优先队列

题目

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例 1：

输入：lists = [[1,4,5],[1,3,4],[2,6]]
输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下：
[ 1->4->5, 1->3->4, 2->6 ] 将它们合并到一个有序链表中得到。 1->1->2->3->4->4->5->6

示例 2：

输入：lists = []
输出：[]

示例 3：

输入：lists = [[]]
输出：[]

解题思路
方法1.分而治之，类似于归并排序。

方法2.优先队列维护 非空的链表头即可。
有小优化就是可以在优先队列取出一个链头后，记录下一个链头的值，在当前链头遍历，小于等于此值的都可以直接加入，而不需要再加入队列中。
【注意】优先队列声明方式中 排序方法需要使用operator方式传入，不能和sort那样可以直接传入cmp函数。
优先队列自定义比较方式可参考：https://blog.csdn.net/AAMahone/article/details/82787184?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.add_param_isCf&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.add_param_isCf
方法2代码

/**
 * 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:
struct cmp{
    bool operator()(ListNode* x,ListNode* y)
    { return x->val > y->val ; }
};

ListNode* mergeKLists(vector<ListNode*>& lists) {
    int len = lists.size();
    if( len == 0) return NULL;
    if(len == 1)
    {
        return lists[0];
    }
    priority_queue<ListNode*, vector<ListNode*>, cmp > heads;
    for(int i=0;i<len;i++)
    {
        if(lists[i] != NULL )
        {
            heads.push(lists[i]);
        }
       
    }
    ListNode* ans = new ListNode(0);
    ListNode* cur = new ListNode(0);
    cur = ans;
    while( !heads.empty())
    {
        ListNode* temp = heads.top();
        heads.pop();
        int minn = heads.top()->val;
        while( temp!=NULL && temp->val<=minn)
        {
            cur->next = temp;
            cur = cur->next;
            temp=temp->next;
        }
        if(temp!=NULL )
        {
            heads.push(temp);
        }

    }
    return ans->next;
}
};