LeetCode 23. 合併K個排序連結串列

畫船聽雨發表於2018-07-09

題目描述

這裡寫圖片描述

解題思路

堆排序

將每個連結串列中元素放在一起,構造一個元素個數為k的小頂堆,每次取出堆頂元素,構成新的連結串列。
並把該元素後面的元素加入到堆中,反覆操作直到堆為空。
使用優先佇列來實現堆,可以方便運算。

程式碼實現

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
struct cmp
{
    bool operator () (ListNode *a, ListNode *b)
    {
        return a->val > b->val;
    }
};


class Solution
{
public:
    ListNode* mergeKLists(vector<ListNode*>& lists)
    {
        priority_queue<ListNode*, vector<ListNode*>, cmp> que;
        for(int i = 0; i < lists.size(); i++)
        {
            if(lists[i])
                que.push(lists[i]);
        }
        ListNode *head = NULL;
        ListNode *pre = NULL;
        ListNode *temp = NULL;
        while(!que.empty())
        {
            temp = que.top();
            que.pop();
            if(!pre) head = temp;
            else pre->next = temp;
            pre = temp;
            if(temp->next) que.push(temp->next);
        }
        return head;
    }
};

相關文章