力扣 23. 合并K个升序链表顺序合并

Elvin域文

已于 2022-10-28 10:34:38 修改

阅读量631

点赞数 1

分类专栏：算法文章标签： leetcode 链表算法

于 2022-10-27 12:04:18 首次发布

本文链接：https://blog.csdn.net/m0_51787573/article/details/127549179

版权

算法专栏收录该内容

31 篇文章 0 订阅

订阅专栏

第一站 LeetCode 新手村

前言

最近玩OJ赛，发现对算法的理解还需要更加扎实，code能力还可以进一步提升，所以做这样一个算法的系列文章，用于记录学习心得，交流经验，更好地进步和成长。

23. 合并K个升序链表

题目描述

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例1

输入：lists = [[1,4,5],[1,3,4],[2,6]]
输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下：
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6

示例 2

输入：lists = []
输出：[]

示例 3

输入：lists = [[]]
输出：[]

提示

k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i] 按升序排列
lists[i].length 的总和不超过 10^4

解题思路

预知

LeetCode是核心代码模式，所以只需要考虑核心算法，输入由系统自动完成，最后的输出以return返回；

思路

思路：顺序合并

链表中的元素两两合并，新遍历到的链表和已经合并好的列表进行比对，每次合并新旧两个链表，具体思路请参考代码注释，结合代码，理解更深刻；

力扣官方题解放在下方，感兴趣的小伙伴可以去了解下分冶的解法

力扣

代码

C++

元素两两合并

/**
 * 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) {}
 * };
 */

// 思路：链表中的元素两两合并，新遍历到的链表和已经合并好的列表进行比对，每次合并两个链表
// 时间复杂度：由于要不断遍历，排序，O(K²n)
// 空间复杂度：并未用到新的结果链表长度规模的空间，所以空间复杂度为O（1）
class Solution {
public:
    ListNode* mergeTwoLists(ListNode* list1, ListNode* list2){
        if(!list1) return list2;      //若list1为空，则输出list2
        if(!list2) return list1;      //若list2为空，则输出list1    
                                      //若两个链表均为空，则输出null

        ListNode head; //声明一个虚拟头结点，head的val属性不保存任何值
        ListNode* tail = &head, *list1_ptr = list1, *list2_ptr=list2;
        //tail代表尾节点，初始时指向头节点，后逐步添加元素     
        //*list1_ptr, *list2_ptr 指向list1, list2的头结点

        while(list1_ptr and list2_ptr){ //当链表均不为空时
            if(list1_ptr->val<list2_ptr->val){   
                tail->next = list1_ptr;    //尾指针指向小的元素，
                list1_ptr = list1_ptr->next;  //较小元素的链表指针向后遍历，寻找下一个值继续于较大元素比较
            }
            else{
                tail->next = list2_ptr;
                list2_ptr = list2_ptr->next;
            }
            tail = tail->next; //新元素加入后，更新尾结点，指向最后一个元素
        }
        tail->next = (list1_ptr ? list1_ptr : list2_ptr);
        return head.next; //不能直接使用tail输出结果的原因是，在操作的过程中tail始终指向尾部
    }                     //若链表想全部输出则必须从头结点开始遍历，逐个输出

    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode* res = nullptr;   //声明结果指针
        for(int i=0;i<lists.size();i++){  //遍历链表数组，逐个添加，两两比较
            res = mergeTwoLists(res, lists[i]);
        }
        return res;  //返回结果
    }
};