LeetCode23 合并K个排序链表

最新推荐文章于 2022-01-18 22:18:37 发布

C.R.A.zy

最新推荐文章于 2022-01-18 22:18:37 发布

阅读量152

点赞数

文章标签： leetcode

本文链接：https://blog.csdn.net/weixin_43350346/article/details/104830338

版权

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

示例:

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

class Solution {
public:
    struct cmp
    {  
       bool operator()(ListNode *a,ListNode *b){
          return a->val > b->val; }
    };
    ListNode* mergeKLists(vector<ListNode*>& lists) {
       priority_queue<ListNode* ,vector<ListNode*> , cmp> heapk;
       for(auto p:lists)
   {
        if(p!=NULL)
        {
            heapk.push(p);
        }
   }
       ListNode *pHead = new ListNode(-1);
       ListNode *pCur = pHead;
       while(!heapk.empty())
       {
          ListNode *top = heapk.top();heapk.pop();
          pCur->next = top;
          pCur = pCur->next;
          if(top->next!=NULL)
          {
            heapk.push(top->next);
          } 
       }
       pCur = pHead->next;
       delete pHead;
       return pCur;
    }
};