Leedcode编程题23： 合并K个排序链表----C++实现

目的

旨在记录在Leedcode网上刷题的过程，记录心得。

题目

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

思路

参考大神的思路，实现的代码。

1）获取到所有的链表元素放在vector中；

2）将vector从小到大排序；

3）根据排好序的vector元素来创建链表，所得链表即为结果。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        vector<int> templist;
        for(auto li:lists){
            while(li){
                templist.push_back(li->val);
                li = li->next;
            }
        }
        int numsize = templist.size();
        if(numsize==0) return NULL;
        else{
            sort(templist.begin(),templist.end());
            ListNode* resultptrhead = new ListNode(templist[0]);
            ListNode *p = NULL,*q = NULL;
            for(int i=1;i<numsize;i++){
                p = new ListNode(templist[i]);
                p->next = NULL;
                if(!resultptrhead->next){//若head->next为NULL,则
                    resultptrhead->next = p;
                    q = p;
                }else{
                    q->next = p;
                    q = p;
                }
            }
            return resultptrhead;
        }
    }
};

效果