13.Merge k Sorted Lists

最新推荐文章于 2024-04-03 21:24:25 发布

转载最新推荐文章于 2024-04-03 21:24:25 发布 · 55 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/freeopen/p/5483010.html

本文介绍了一种使用映射与链表合并的高效算法，通过映射值和链表指针，实现时间复杂度为O(N)，空间复杂度也为O(N)的优化。详细解释了算法步骤，并提供了代码实现。

    思路：利用map<int,vector<ListNode*> >
做值和指针的映射，最后将指针按值依次链接起来，
时间复杂度O(N),空间O(N)    
    Merge k sorted linked lists and
return it as one sorted list. Analyze and describe its complexity.

/**
 * 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) {
        map<int,vector<ListNode*> > maps;
        for(int i=0;i<lists.size();++i)
        {
            while(lists[i]!=NULL)
            {
                maps[lists[i]->val].push_back(lists[i]);
                lists[i] = lists[i]->next;
            }
        }
        ListNode *now=NULL,*pre=NULL,*head;
        map<int,vector<ListNode*> >::const_iterator it = maps.begin();
        if(it!=maps.end()){
                head = (it->second)[0];
        }
        else head = NULL;
        while(it!=maps.end())
        {
            cout<<(it->second).size()<<endl;
            for(int k=0;k<(it->second).size();++k)
            {
                pre = now;
                now = (it->second)[k];
                if(pre!=NULL)pre->next = now;
            }
            it++;
        }
        return head;
    }
};