23. Merge k Sorted Lists

最新推荐文章于 2024-01-12 14:30:52 发布

zibingling777

最新推荐文章于 2024-01-12 14:30:52 发布

阅读量77

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题笔记文章标签： Linked List

本文链接：https://blog.youkuaiyun.com/zibingling777/article/details/78825045

刷题笔记专栏收录该内容

304 篇文章

订阅专栏

本文介绍了一种将多个已排序的链表合并为一个有序链表的方法。提供了两种实现方式：一种是通过两两合并的方式逐步减少链表数量直至只剩一个；另一种则是使用优先队列（最小堆）来高效地选取最小元素并构建最终链表。

1、题目描述

给了k个有序链表。返回合并成的一个链表。

2、思路

两个两个合并，合并到只剩下一个返回。

思路2：优先队列（即堆排序）。

3、代码

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0) return NULL;
        while(lists.size()>1){
            lists.push_back(merge(lists[0],lists[1]));
            lists.erase(lists.begin());
            lists.erase(lists.begin());
        }
        return lists[0];
    }
     ListNode* merge(ListNode* a,ListNode* b){
         ListNode* ans = new ListNode(-1);
         ListNode* p =ans;
         while(a&&b){
             if(a->val<b->val){
                 ans->next = a;
                 a=a->next;
                 ans=ans->next;
             }
             else{
                 ans->next = b;
                 b=b->next;
                 ans=ans->next;
             }
         }
         if(a)
             ans->next = a;
         if(b)
             ans->next = b;
         return p->next;
     }

struct cmp{  
   bool operator()(ListNode* a,ListNode* b){  
       return a->val > b->val;  
   }   
};  
class Solution {
public:
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        if(lists.size() == 0)
            return NULL;
        priority_queue<ListNode*, vector<ListNode*>, cmp> queue;
        for(int i = 0; i < lists.size(); i++) {
            if(lists[i] != NULL)
            queue.push(lists[i]);
        }
        ListNode * head = new ListNode(-1);
        ListNode * ans = head;
        while(queue.size() > 0) {
            ListNode * t = queue.top();
            queue.pop();
            head->next = t;
            head = head->next;
            if(t->next)
                queue.push(t->next);
        }
        return ans->next;    
    }
};