合并两个有序链表

本文介绍了如何合并两个有序链表,分别提供了递归和迭代两种解法。递归解法通过比较两个链表头节点的值,进行合并操作,空间复杂度为O(n+m)。迭代解法利用哨兵节点和额外指针,逐个比较节点并合并,空间复杂度为O(1)。

21. 合并两个有序链表

题目描述:

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

来源:力扣(LeetCode)21
链接:https://leetcode-cn.com/problems/find-words-that-can-be-formed-by-characters

示例:

输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4

题解

方法一:递归

思路

我们可以如下递归地定义两个链表里的 merge 操作(忽略边界情况,比如空链表等):

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

我们直接将以上递归过程建模,同时需要考虑边界情况。

我们要先判断给定 两个链表是否为空链表 ,那么没有任何操作需要合并,所以我们只需要返回非空链表。否则,我们要判断 l1l2 头结点哪个更小,然后较小结点的 next 指针指向其余结点的合并结果。(调用递归)如果两个链表有一个为空,递归结束。

示例代码

/**
​
 \* 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* l1, ListNode* l2) {
        
        // 先判断给定链表是否为空
        if(l1 == nullptr)
            return l2;
        else if (l2 == nullptr)
            return l1;
        
        // 不为空,判断l1和l2头结点哪个更小,然后较小结点的next指针指向其余结点的合并结果
        else if (l1->val < l2->val)
        {
            l1->next = mergeTwoLists(l1->next, l2);
            return l1;
        }          
        else 
        {
            l2->next = mergeTwoLists(l2->next, l1);
            return l2;
        }
           
    }
};

复杂度分析

  • 时间复杂度:O(n+m),其中 n 和 m 分别为两个链表的长度。因为每次调用递归都会去掉 l1 或者 l2 的头节点(直到至少有一个链表为空),函数 mergeTwoLists 至多只会递归调用每个节点一次。因此,时间复杂度取决于合并后的链表长度,即 O(n+m)。

  • 空间复杂度:O(n+m),其中 n 和 m 分别为两个链表的长度。递归调用 mergeTwoLists 函数时需要消耗栈空间,栈空间的大小取决于递归调用的深度。结束递归调用时 mergeTwoLists 函数最多调用 n+m次,因此空间复杂度为 O(n+m)。

 

方法二: 迭代

思路

迭代的话,我们要两个链表中一个一个元素拿出来比,所以为了操作方便,我们可以另建链表,然后依次将较小元素追加到新建链表后边。为了比较完之后知道要将元素插在哪里,所以要记录一下最后插入元素的位置。

首先,我们设定一个哨兵节点 head ,这可以在最后让我们比较容易地返回合并后的链表。

然后维护一个 prev 指针,来记录最后插入元素的位置,方便L1、L2比较完之后快速插入。我们需要做的是调整它的 next 指针。

我们重复以下过程,直到 l1 或者 l2 中没有元素。如果 l1 当前节点的值小于等于 l2 ,我们就把 l1 当前的节点追加在 prev 节点的后面同时将 l1 指针往后移一位。否则,我们对 l2 做同样的操作。不管我们将哪一个元素追加在新链表后面,我们都需要移动 prev 的位置。

在循环终止的时候, 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* l1, ListNode* l2) {
​
        ListNode *head = new ListNode();
        ListNode *preNode = head;
​
        // 循环至有链表元素为空
        while(l1 != nullptr && l2 != nullptr)
        {
            if(l1->val < l2->val)
            {
                preNode->next = l1;
                l1 = l1->next;
            }          
            else
            {
                preNode->next = l2;
                l2 = l2->next;
            }    
            // 插入一个元素,记录指针后移
            preNode = preNode->next;  
        }
        
        // 找剩下的非空链表,将其剩余元素全部追加到新链表中
        l1 == nullptr ? preNode->next = l2 : preNode->next = l1;
           
        return head->next;
    }
};

复杂度分析

  • 时间复杂度:O(n+m),其中 n 和 m 分别为两个链表的长度。因为每次循环迭代中,l1 和 l2 只有一个元素会被放进合并链表中, 因此 while 循环的次数不会超过两个链表的长度之和。所有其他操作的时间复杂度都是常数级别的,因此总的时间复杂度为 O(n+m)。

  • 空间复杂度:O(1)。我们只需要常数的空间存放若干变量。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值