首页 > 其他 > 详细

Leetcode 23. Merge k Sorted Lists

时间:2019-05-08 17:58:01      阅读:136      评论:0      收藏:0      [点我收藏+]

https://leetcode.com/problems/merge-k-sorted-lists/

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

#include <queue>
using namespace std;
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        /*
        
        把每一个链表的头指针放到优先队列中,每次选最小的那个,取出后将其下一个指针(不为空)放入优先队列.
        边界:
        1.有些链表开始就为空
        
        */
        struct cmp{
            bool operator() (ListNode *a,ListNode *b){return a->val>b->val;};
        };
        
        ListNode* pre=new ListNode(0),*pp=pre;
        priority_queue<ListNode*,vector<ListNode*>,cmp> que; 
        for(auto& p:lists)
            if(p) que.push(p);
        while(!que.empty()){
            //取最小的
            auto *p=que.top();
            que.pop();
            
            pp->next=p;
            pp=pp->next;
            //加入新的指针
            if(pp->next) que.push(pp->next);
        }
        pp=pre->next;
        delete pre;
        return pp;
        
    }
};

Leetcode 23. Merge k Sorted Lists

原文:https://www.cnblogs.com/ximelon/p/10832798.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!