Leetcode Merge k Sorted Lists 利用stl heap来实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
struct Greater {
  bool operator() (ListNode const * const x, ListNode const * const y) const {return x && (!y || x->val > y->val);}
  //只要记住const只对其左边的操作符施加作用,因此上述表达式翻译为
  //ListNode const * x 代表x指针指向可以改,但是指向的数值不能改
  //ListNode* const x  代表x指针指向不能改,但是指向的数值可以改
  //ListNode const * const x 代表x指针以及所指向的数值都不能改
  //任何不会修改数据成员的函数都因该声明为const类型。在编写const成员函数时,不慎修改了数据成员,或者调用了其他非const成员函数,编译器将报错 
};
class Solution {
public:
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        if(lists.size()<=0){
           return NULL;
        }
        make_heap(lists.begin(), lists.end(), Greater());
        ListNode *head = NULL, *curr = head;
        while(!lists.empty()){
            if(head == NULL){
                head = curr = lists.front();
            }else{
                curr->next = lists.front();
                curr = curr -> next;
            }
            pop_heap(lists.begin(), lists.end(), Greater());
            if(lists.back()){
                ListNode *tmp = lists.back()->next;
                if(tmp){
                    lists.back() = tmp;
                    push_heap(lists.begin(),lists.end(),Greater());
                }else{
                    lists.pop_back();
                }
            }else{
                lists.pop_back();
            }
        }
        return head;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值