題目

給你一個連結串列陣列，每個連結串列都已經按升序排列。

請你將所有連結串列合併到一個升序連結串列中，返回合併後的連結串列。

示例一

輸入：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

思路分析

• 使用優先佇列模擬小頂堆，每次取隊頭元素，作為最小的元素。

程式碼

class Solution {
public:
    struct cmp{
        bool operator()(ListNode * const &a, ListNode * const &b){//按照降序排列
            return a->val > b->val;
        }
    };
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        int n = lists.size();
        ListNode head = ListNode(0);
        priority_queue<ListNode *, vector<ListNode *>, cmp> q;
        for(auto p : lists) if(p != NULL) q.push(p);
        ListNode * p = &head;
        while(!q.empty()){
            p->next = q.top();
            q.pop();
            p = p->next;
            if(p->next) q.push(p->next);
        }
        return head.next;
    }
};