LeetCode 23： Merge k sorted Lists

现在已将LeetCode刷题心得记录github上，每一题都有我能够想到的多种解法，欢迎star！github   

 

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

Example:

Input:
[
  1->4->5,
  1->3->4,
  2->6
]
Output: 1->1->2->3->4->4->5->6

 

一开始最简单的想法就是利用k个指针分别指向k个List的当前位置，每次遍历k个指针寻找最小值并新建结点，但是这样的方式时间效率非常的低下，理论分析来说 O(nk) ，当k=n的时候也就O（n2），不知道为什么会这么慢。。

/**
 * 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<ListNode*>& point;
        // for(int i=0;i<lists.size();i++)
        //     if(lists[i]!=NULL)
        //         point.push_back(lists[i]);
        
        ListNode*start=NULL,*end=NULL;
        int flag=1,min=-1;
        while(flag)
        {
            flag=0;
            min=-1;
            for(int i=0;i<lists.size();i++)
            {
                if(lists[i]!=NULL)
                {
                    flag=1;
                    if(min==-1) min=i;
                    min=lists[i]->val<lists[min]->val?i:min;
                }
            }
            if(min==-1) continue;
            if(start==NULL)
            {
                start=new ListNode(lists[min]->val);
                end=start;
            }
            else{
                end->next=new ListNode(lists[min]->val);
                end=end->next;
            }
            lists[min]=lists[min]->next;
        }
        return start;
    }
};

   接着考虑牺牲空间换时间，将所有元素放到数组，排序之后再新建到链表里面，这样只需要O(nlgn+2n)左右的代价，此处可以使用快速排序或者堆排序，时间能从484ms降低到16ms

 

/**
 * 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>res;
		for (int i = 0; i < lists.size(); i++)
		{
			while (lists[i] != NULL)
			{
				res.push_back(lists[i]->val);
				lists[i] = lists[i]->next;
			}
		}
		sort(res.begin(), res.end());
		if (res.empty()) return NULL;
		ListNode* start = new ListNode(res[0]);
		ListNode* end = start;

		for (int i = 1; i < res.size(); i++)
		{
			end->next = new ListNode(res[i]);
			end = end->next;
		}
		return start;
    }
};