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

一曲冷凌霜 提交于 2020-02-04 11:26:38

目的

旨在记录在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;
        }
    }
};

效果

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!