leetcode 合并k个排序链表

最新推荐文章于 2020-08-11 14:51:08 发布

原创最新推荐文章于 2020-08-11 14:51:08 发布 · 210 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

leetcode 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用暴力方法合并K个已排序链表的算法，并提供了详细的代码实现过程。当输入的链表列表为空时，算法直接返回NULL；否则，通过不断合并相邻链表直至只剩下一个链表。

1.暴力通过128/137个样例，最终超时

当lists为空时，输出NULL即可。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {

public: 
    void merge(vector<ListNode*>& list,int s,int e){
        
        //合并两个链表
        ListNode* templ=new ListNode(-1);
        ListNode* l1=list[s];
        ListNode* l2=list[e];
        ListNode* ans=templ;
        while(l1!=NULL&&l2!=NULL){
            cout<<l1->val<<" "<<l2->val<<endl;
            if(l1->val<l2->val){
                ListNode* temp= new ListNode(l1->val);
                templ->next=temp;
                l1=l1->next;
                templ=templ->next;
                cout<<templ->val<<endl;
            }else{
                ListNode* temp= new ListNode(l2->val);
                templ->next=temp;
                l2=l2->next;
                templ=templ->next;
                cout<<templ->val<<endl;
            }
        }
        while(l1!=NULL){
            ListNode* temp= new ListNode(l1->val);
            templ->next=temp;
            l1=l1->next;
            templ=templ->next;
            cout<<templ->val<<endl;
        }
        while(l2!=NULL){
            ListNode* temp= new ListNode(l2->val);
            templ->next=temp;
            l2=l2->next;
            templ=templ->next;
            cout<<templ->val<<endl;
        }
        list[s]=ans->next;
    }
    
    
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0)return NULL ;
        if(lists.size()==1)return lists[0];
        else{
            cout<<lists.size()<<endl;
            for(int i=lists.size()-2;i>=0;i--){
                merge(lists,i,i+1);
            }
            return lists[0];
        }
    }
};

2.优先队列法