Merge K sorted Lists

最新推荐文章于 2021-02-02 21:57:42 发布

L_J_SHOU

最新推荐文章于 2021-02-02 21:57:42 发布

阅读量2.3k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： Algo. and Data structure 文章标签： leetcode merge-k-sorted-lists

本文链接：https://blog.youkuaiyun.com/shoulinjun/article/details/19576585

Algo. and Data structure 专栏收录该内容

57 篇文章

订阅专栏

本文介绍了一种使用最小堆合并k个已排序链表的方法，并分析了其复杂度。通过堆操作，我们可以高效地实现链表合并。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

采用最小堆，首先将 k 个首节点放入堆中，弹出最小的节点并插入到新的链表中；

弹出的节点如果next 非空，就将它的下一个节点进堆。

继续，直到堆为空。

堆 Push 和 pop 的复杂度是 log(k), 所以复杂度是 nlg(k), n为总的节点数；

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        int K = lists.size();
        if (K == 0) return NULL;
        else if (K == 1) return lists[0];
        
        ListNode *listHead(NULL), *listRear(NULL);
        ListNode *node(NULL);
        priority_queue<ListNode*, vector<ListNode*>, cmp> h;
        
        // push K list heads into heap
        for(int i=0; i<K; ++i)
          if(lists[i] != NULL){
            h.push(lists[i]);
            lists[i] = lists[i]->next;
          }
            
        while(!h.empty()){
          //pop the min of k nodes
          node = h.top(); h.pop();
          if(node->next)
            h.push(node->next);
            
          //insert node into new list    
          if(listRear){
              listRear->next = node;
              listRear = listRear->next;
          }
          else{
              listHead = listRear = node;
          }
        }
        
        return listHead;
    }
private:
    struct cmp{
        bool operator()(ListNode* lhs, ListNode *rhs){
            if(lhs->val < rhs->val) return false;
            else return true;
        }
    };
};