Leetcode（21）——合并两个有序链表

SmileGuy17

已于 2022-05-22 09:46:58 修改

阅读量300

点赞数

分类专栏： Leetcode 文章标签：链表 leetcode c++

于 2022-05-22 09:46:49 首次发布

本文链接：https://blog.csdn.net/KCDCY/article/details/124906757

版权

Leetcode 专栏收录该内容

77 篇文章 6 订阅

订阅专栏

本文介绍了如何使用迭代和递归两种方法解决LeetCode第21题——合并两个有序链表。首先，详细解释了迭代法的思路，即通过比较两个链表的头节点值，将较小值的节点添加到结果链表中，并更新指针。然后，阐述了递归法的实现，同样比较头节点值，但通过递归调用来处理剩余部分。最后，分析了两种方法的时间复杂度均为O(n+m)，空间复杂度分别为O(n+m)和O(1)。

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

Leetcode（21）——合并两个有序链表

题目

将两个升序链表合并为一个新的升序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

示例 1：

输入：l1 = [1,2,4], l2 = [1,3,4]
输出：[1,1,2,3,4,4]

示例 2：

输入：l1 = [], l2 = []
输出：[]

示例 3：

输入：l1 = [], l2 = [0]
输出：[0]

提示：

两个链表的节点数目范围是 [0, 50]
-100 <= Node.val <= 100
l1 和 l2 均按非递减顺序排列

题解

方法：迭代

思路

我们可以如下递归地定义两个链表里的 merge 操作（忽略边界情况，比如空链表等）：
$\left\{ \begin{array}{ll} list1[0] + merge(list1[1:], list2) & list1[0] < list2[0] \\ list2[0] + merge(list1, list2[1:]) & otherwise \end{array} \right.$

也就是说，两个链表头部值较小的一个节点与剩下元素的 merge 操作结果合并。

代码实现

/**
 * 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* list1, ListNode* list2) {
        if(list1 == nullptr && list2 == nullptr) return nullptr;
        else if(list1 == nullptr) return list2;
        else if(list2 == nullptr) return list1;

        if(list1->val <= list2->val){
            list1->next = mergeTwoLists(list1->next, list2);
            return list1;
        }else{
            list2->next = mergeTwoLists(list1, list2->next);
            return list2;
        }
    }
};

复杂度分析

时间复杂度： $O (n + m)$ ，其中 $n$ 和 $m$ 分别为两个链表的长度。因为每次调用递归都会去掉 l1 或者 l2 的头节点（直到至少有一个链表为空），函数 mergeTwoList 至多只会递归调用每个节点一次。因此，时间复杂度取决于合并后的链表长度，即 $O (n + m)$ 。
空间复杂度： $O (n + m)$ ，其中 $n$ 和 $m$ 分别为两个链表的长度。递归调用 mergeTwoLists 函数时需要消耗栈空间，栈空间的大小取决于递归调用的深度。结束递归调用时 mergeTwoLists 函数最多调用 $n + m$ 次，因此空间复杂度为 $O (n + m)$ 。

方法二：递归

思路

我们可以用迭代的方法来实现上述算法。当 l1 和 l2 都不是空链表时，判断 l1 和 l2 哪一个链表的头节点的值更小，将较小值的节点添加到结果里，当一个节点被添加到结果里之后，将对应链表中的节点向后移一位。

代码实现

/**
 * 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* list1, ListNode* list2) {
        if(list1 == nullptr && list2 == nullptr) return nullptr;
        else if(list1 == nullptr) return list2;
        else if(list2 == nullptr) return list1;

        ListNode* ans = nullptr,* tmp = nullptr;
        while(list1 != nullptr && list2 != nullptr){
            if(list1->val <= list2->val){
                if(ans == nullptr){
                    ans = list1;
                    tmp = list1;
                }else{
                    tmp->next = list1;
                    tmp = tmp->next;
                }
                list1 = list1->next;
            }else{
                if(ans == nullptr){
                    ans = list2;
                    tmp = list2;
                }else{
                    tmp->next = list2;
                    tmp = tmp->next;
                }
                list2 = list2->next;
            }
        }
        tmp->next = list1 == nullptr? list2: list1;
        return ans;
    }
};

复杂度分析

时间复杂度： $O (n + m)$ ，其中 $n$ 和 $m$ 分别为两个链表的长度。因为每次循环迭代中，l1 和 l2 只有一个元素会被放进合并链表中，因此 while 循环的次数不会超过两个链表的长度之和。所有其他操作的时间复杂度都是常数级别的，因此总的时间复杂度为 $O (n + m)$ 。
空间复杂度： $O (1)$ ，我们只需要常数的空间存放若干变量。