创建优先级队列,将牛群中的每个节点放入优先级队列中,然后依次弹出优先级队列中的值组合成结果链表返回。
/**
* struct ListNode {
* int val;
* struct ListNode *next;
* ListNode(int x) : val(x), next(nullptr) {}
* };
*/
#include <queue>
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param head ListNode类
* @return ListNode类
*/
ListNode* sortList(ListNode* head) {
// write code here
if(head == nullptr) {
return head;
}
priority_queue<ListNode*, vector<ListNode*>, Compare> pri_queue;
ListNode* current = head;
while(current != nullptr) {
pri_queue.push(current);
current = current->next;
}
ListNode* result = pri_queue.top();
pri_queue.pop();
current = result;
while(pri_queue.size() != 0) {
current->next = pri_queue.top();
pri_queue.pop();
current = current->next;
if(pri_queue.size() == 0) {
current->next = nullptr;
}
}
return result;
}
struct Compare {
bool operator()(const ListNode* a, const ListNode* b) {
return a->val > b->val;
}
};
};



京公网安备 11010502036488号