LeetCode 023 Merge k Sorted Lists

题目要求：Merge k Sorted Lists

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

 

分析：

参考网址：http://blog.csdn.net/a83610312/article/details/8554241

思路是：首先将k个链表的第一个节点集合，建堆，然后取出堆顶的节点，链接到结果链表上，然后将该节点的下一个节点入堆，直到所有链表都已经完成；

下面的Node结构体好像根本无必要，刚开始没注意是链表（而链表可以很方便的获取下一个节点，如果是数组的话还得记录它来自哪个数组）；

有两点需要注意：

1.当一条链表结束的时候，添加值为INF的新Node加入堆，可以方便地避免NULL的判断；

2.make_heap()默认是大顶堆，所以要显示指定使用greater<node>()，来获取小顶堆；

 

代码如下：

#define INF 1000000



struct node{

    

    int val;

    ListNode* from;

    

    node(ListNode* n){

        if(n == NULL) val = INF;

        else val = n->val;

        from = n;

    }



    bool operator<(const node& other)const{

        return val<other.val;

    }

    

    bool operator>(const node& other)const{

        return val>other.val;

    }

};

    

class Solution {

public:

    ListNode *mergeKLists(vector<ListNode *> &lists) {

        // Start typing your C/C++ solution below

        // DO NOT write int main() function

        

        if (lists.empty()) return NULL;

        int n = lists.size();

        vector<node> heap;

        heap.reserve(n);

        

        for(int i = 0; i < n; i++)

            heap.push_back(node(lists[i]));

            

        make_heap(heap.begin(), heap.end(),greater<node>());

        ListNode* head= new ListNode(0);

        ListNode* pL = head;

        pop_heap(heap.begin(), heap.end(),greater<node>());

        node small = heap.back();

        heap.pop_back();

        

        while(small.val!=INF){

            

            ListNode* next=small.from->next;

            pL->next=small.from;

            small.from->next=NULL;

            pL=pL->next;

    

            heap.push_back(node(next));

            push_heap(heap.begin(), heap.end(),greater<node>());

            pop_heap(heap.begin(), heap.end(),greater<node>());

            small=heap.back();

            heap.pop_back();

        }

    

        ListNode* ret=head->next;

        delete head;

        return ret;    

        

    }

};