创建优先级队列,将牛群中的每个节点放入优先级队列中,然后依次弹出优先级队列中的值组合成结果链表返回。
/** * 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; } }; };