23. 合并K个升序链表（顺序合并/分治合并、c++）

程序媛JD

已于 2022-01-28 16:54:06 修改

阅读量699

点赞数

分类专栏：力扣刷题笔记文章标签：链表 c++ leetcode

于 2022-01-28 16:52:38 首次发布

本文链接：https://blog.csdn.net/qq_36421001/article/details/122733268

版权

力扣刷题笔记专栏收录该内容

221 篇文章 6 订阅

订阅专栏

在这里插入图片描述

方法一：顺序合并

思路

我们可以想到一种最朴素的方法：用一个变量 ans 来维护以及合并的链表，第 i 次循环把第 i个链表和 ans 合并，答案保存到 ans 中。

/**
 * 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* mergeTwoList(ListNode* a,ListNode* b){
        if((!a) || (!b)){
            return a ? a : b;//a或者b为空时，返回不为空的链表
        }
        ListNode head, *tail = &head,*p1 = a,*p2 = b;
        while(p1 && p2){
            if(p1->val > p2->val){
                tail->next = p2;
                p2 = p2->next;
            }else{
                tail->next = p1;
                p1 = p1->next;
            }
            tail = tail -> next;
        }
        tail -> next = (p1?p1:p2);
        return head.next;
    }
    //顺序合并
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode* ans = nullptr;
        for(int i = 0; i < lists.size(); ++i){//int i = 0 <=> size_t i = 0;
            ans = mergeTwoList(ans, lists[i]);//顺序合并
        }
        return ans;

    }
};

在这里插入图片描述

/**
 * 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* mergeTwoList(ListNode* a,ListNode* b){
        if((!a) || (!b)){
            return a ? a : b;//a或者b为空时，返回不为空的链表
        }
        ListNode head, *tail = &head,*p1 = a,*p2 = b;
        while(p1 && p2){
            if(p1->val > p2->val){
                tail->next = p2;
                p2 = p2->next;
            }else{
                tail->next = p1;
                p1 = p1->next;
            }
            tail = tail -> next;
        }
        tail -> next = (p1?p1:p2);
        return head.next;
    }
    
    //分治
    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;//相当于/2
        return mergeTwoList(merge(lists, l, mid),merge(lists, mid + 1, r));
    } 

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        return merge(lists, 0, lists.size() - 1);
    }
};