合并K个有序链表

qq_45576085

已于 2024-03-04 19:08:07 修改

阅读量354

点赞数 7

分类专栏：刷题文章标签：链表数据结构

于 2024-03-03 17:17:19 首次发布

本文链接：https://blog.csdn.net/qq_45576085/article/details/136434342

版权

刷题专栏收录该内容

29 篇文章 0 订阅

订阅专栏

题目

题目链接

合并k个已排序的链表_牛客题霸_牛客网

题目描述

代码实现

class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param lists ListNode类vector
     * @return ListNode类
     */
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        // write code here
        return DivideList(lists, 0, lists.size() - 1);
    }
    ListNode* DivideList(vector<ListNode*>& lists, int left, int right) {
        if (left > right)
            return nullptr;
        //中间只有一个链表的情况
        if (left == right)
            return lists[right];
        //从中间分为两段，在将两段合并好的链表再合并
        int mid = (left + right) / 2;
        return MergeList(DivideList(lists, left, mid), DivideList(lists, mid + 1,right));
    }
    ListNode* MergeList(ListNode* head1, ListNode* head2) {
        if (head1 == nullptr)
            return head2;
        if (head2 == nullptr)
            return head1;
        ListNode* phead = new ListNode(0);
        ListNode* cur = phead;
        while (head1 && head2) {
            if (head1->val <= head2->val) {
                cur->next = head1;
                head1 = head1->next;
            } else {
                cur->next = head2;
                head2 = head2->next;
            }
            cur = cur->next;
        }
        cur->next = head1 == nullptr ? head2 : head1;
        return phead->next;
    }
};