Day12-Task23.合并K个排序链表

و✧ A

于 2020-03-12 19:14:34 发布

阅读量247

点赞数

CC 4.0 BY-SA版权

文章标签：算法 leetcode

本文链接：https://blog.csdn.net/NewInt_HAONAN/article/details/104823946

题目描述

23.合并K个排序链表

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

示例

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

思路

1.暴力法

将链表中所有元素加入到一个数组中
对数组进行排序
新建链表，将数组中元素放入链表中。
时间复杂度 O(nlogn) ,***n***为链表中总结点个数
所需空间 O(n) ，新建链表 O(n) ，数组空间 O(n)

代码如下

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        vector<int> ans;
        for(int i = 0; i < lists.size(); i++)
        {
            while(lists[i] != NULL)
            {
                ans.push_back(lists[i]->val);
                lists[i] = lists[i]->next;
            }
        }
        sort(ans.begin(), ans.end());
        ListNode* p = new ListNode(0);
        ListNode* q = p;
        for(int i = 0; i <ans.size(); i++)
        {
            p->next = new ListNode(ans[i]);
            p = p->next;
        }
        return q->next;
    }
};