LeetCode——第二十二天（合并K个排序链表）

LeetCode——第二十二天

23. 合并K个排序链表

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

分析：只会暴力，看了官方分治，还有一个优先队列就比较好玩了。分治我没有去写，这边写下优先队列，其实也挺简单，就是比如例题，因为本身都是排序好的，优先队列会维护自己排序，然后把最小的pop后加入最后链表，被选中的那个就直接next，重复就好了。不过简单的原因也是没有自己写优先队列，自己写一个还是挺麻烦的。

优先队列

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    struct Status {
        int val;
        ListNode *ptr;
        //重置<运算符
        bool operator < (const Status &rhs) const {
            return val > rhs.val;
        }
    };

    priority_queue <Status> q;//优先队列

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        for (auto node: lists) {
            if (node) q.push({node->val, node});//头结点按顺序加入
        }
        ListNode head, *tail = &head;
        while (!q.empty()) {
            auto f = q.top(); q.pop();//弹出最小值,加入tail链表中
            tail->next = f.ptr; 
            tail = tail->next;
            if (f.ptr->next) q.push({f.ptr->next->val, f.ptr->next});//当前退出值下一个如果非空则加入优先队列中
        }
        return head.next;
    }
};

2.总结