Merge k Sorted Lists

最新推荐文章于 2021-12-08 15:59:42 发布

weixin_30783913

最新推荐文章于 2021-12-08 15:59:42 发布

阅读量56

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/summer-zhou/p/3359538.html

本文介绍了一种使用优先级队列合并K个已排序链表的方法。通过定义比较器来确保链表节点按升序排列，并利用优先级队列不断取出最小元素并加入下一个节点的方式，实现了高效地合并多个链表的功能。

使用优先级队列

struct cmp{
    bool operator() ( ListNode* a, ListNode* b ){
        return a->val>b->val;
    }
};

class Solution {
public:
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        // Note: The Solution object is instaentiated only once and is reused by each test case.
        int k = lists.size();
        priority_queue<ListNode*,vector<ListNode*>,cmp> q;
        for(int i=0;i<k;i++)
        {
            if(lists[i])  //attention, judege if it's NULL
                q.push(lists[i]);
        }
            
        ListNode* newHead = NULL;
        ListNode* tail;
        while(!q.empty())
        {
            ListNode* n = q.top();
            q.pop();
            if(!newHead)
                tail = newHead = n;
            else
            {
                tail->next = n;
                tail = n;
            }
            
            if(n->next)
                q.push(n->next);
        }
        
        return newHead;
    }
};

转载于:https://www.cnblogs.com/summer-zhou/p/3359538.html