23. 合并K个升序链表

最新推荐文章于 2024-01-18 20:36:48 发布

想改行的林

最新推荐文章于 2024-01-18 20:36:48 发布

阅读量194

点赞数

文章标签：链表 list leetcode

本文链接：https://blog.csdn.net/qq_36028321/article/details/122207222

版权

本文探讨了如何通过分治策略优化合并K个升序链表的问题。作者提出了一种递归方法，将链表两两合并，形成树状结构，从而降低了时间复杂度，从O(nV)改进为O(lognV)，其中n为链表总数，V为链表最大长度。这种方法避免了从头到尾遍历vector，提高了算法效率。

摘要由CSDN通过智能技术生成

合并K个升序链表要拆解成多个合并两个升序链表。下面这种写法的时间复杂度还是很高的，如果需要优化，就采用分治的方法（递归）。每次取两个链表合并，变成树形结构，这样时间复杂度要比从头到尾遍历vector要好很多。logn VS n。

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

    ListNode * mergeTwoLists(ListNode *ans, ListNode* list)
    {
        if(ans==nullptr) return list;//针对第一次调用这个函数
        if(list==nullptr) return ans;//针对最后一次调用
        ListNode head;
        ListNode *tail=&head;
        ListNode *p=ans;
        ListNode *q=list;

        while(p&&q)
        {
            if(p->val<q->val)
            {
                tail->next=p;
                p=p->next;
            }
            else
            {
                tail->next=q;
                q=q->next;
            }
            tail=tail->next;
        }
        tail->next=(q?q:p);
        return head.next;
    }

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode *ans=nullptr;//这里用List *ans=new ListNode; ListNode *p=ans;会出现很多问题，暂时想不明白。
        for(int i=0;i<lists.size();++i)
        {
            ans=mergeTwoLists(ans,lists[i]);
        }
        return ans;
    }
};