题解 | #牛群的身高排序#

牛群的身高排序

https://www.nowcoder.com/practice/9ce3d60e478744c09768d1aa256bfdb5

创建优先级队列,将牛群中的每个节点放入优先级队列中,然后依次弹出优先级队列中的值组合成结果链表返回。

/**
 * 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;
        }
    };
};

中等(算法题解) 文章被收录于专栏

中等难度题目

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务