Leecode 23. 合并K个升序链表 链表 递归优先队列

Leecode 23. 合并K个升序链表 链表 递归优先队列,第1张

原题链接:Leecode 23. 合并K个升序链表

解法一:找到每一个链表表头最小值的链表,再递归遍历

class Solution {
public:
    ListNode* root=nullptr;
    ListNode* temp=nullptr;
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0)
            return nullptr;
        int tmp=10010,pos=0,num=0;
        for(int i=0;i<lists.size();i++)
        {
            if(lists[i]!=nullptr && lists[i]->val<=tmp)
            {
                tmp=lists[i]->val;
                pos=i;
            }
            if(lists[i]==nullptr)
                num++;
        }
        if(root==nullptr)
        {
            root=lists[pos];
            temp=root;
        }
        else  if(temp!=nullptr)
        {
            temp->next=lists[pos];
            temp=temp->next;   
        }
        if(lists[pos]!=nullptr)
           lists[pos]=lists[pos]->next;
        if(num!=lists.size())
            mergeKLists(lists);
        return root;
    }
};

解法二:顺序依次合并两个链表

class Solution {
public:
    //合并两个链表
    ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
        if(!list1)
           return list2;
        if(!list2)
           return list1;
        if(list1->val<=list2->val)
        {
            list1->next=mergeTwoLists(list1->next,list2);
            return list1;
        }
        else 
        {
            list2->next=mergeTwoLists(list1,list2->next);
            return list2;
        }
    }

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0)
            return nullptr;
        ListNode* root=lists[0];
        for(int i=1;i<lists.size();i++)
        {
            root=mergeTwoLists(root,lists[i]);
        }
        
        return root;
    }
};

解法三:分治依次合并两个链表


class Solution {
public:
    ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
        if(!list1)
           return list2;
        if(!list2)
           return list1;
        if(list1->val<=list2->val)
        {
            list1->next=mergeTwoLists(list1->next,list2);
            return list1;
        }
        else 
        {
            list2->next=mergeTwoLists(list1,list2->next);
            return list2;
        }
    }
    ListNode* merge(vector<ListNode*>& lists, int l,int r) {
        if(l==r)  return lists[l];
        if(l>r)   return nullptr;
        int mid=(l+r)>>1;
        return mergeTwoLists(merge(lists,l,mid),merge(lists,mid+1,r));
    }
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0)
            return nullptr;
        return merge(lists,0,lists.size()-1);
    }
};

解法四:优先队列

class Solution {
public:
    struct cmp
    {
        bool operator () (const ListNode* a,const ListNode* b) {
            return a->val>b->val;//小顶堆
        }
    };
    //使用优先队列
    priority_queue<ListNode*,vector<ListNode*>,cmp> q;

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0)
            return nullptr;
        for (auto node: lists) {
            if (node) q.push(node);
        }
        ListNode* root=new ListNode();
        ListNode* tmp=root;
        while(!q.empty())
        {
            tmp->next=q.top();
            tmp=tmp->next;
            q.pop();
            if(tmp->next) q.push(tmp->next);
        }
        return root->next;
    }
};

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/565036.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-06
下一篇 2022-04-06

发表评论

登录后才能评论

评论列表(0条)

保存