重写greater函数
重载函数使用结构体实现!!
struct cmp{ //对新的数据类型的 < 进行重写
bool operator()(ListNode *a,ListNode *b){
return a->val > b->val;
}
};
priority_queue<ListNode*, vector<ListNode*>, cmp> heap; // 小根堆
完整代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
struct cmp{ //对新的数据类型的<进行重写
bool operator()(ListNode *a,ListNode *b){
return a->val > b->val;
}
};
ListNode* mergeKLists(vector<ListNode*>& lists) {
pr