瑞文考研算法每日一题—2023.05.25 合并两个有序链表

Problem: 21. 合并两个有序链表

方法一：递归

思路

链表的题目往往可以用递归去思考，递归问题的思考可以拆分成两个部分去思考，即递和归。

• 递：将原问题拆分成子问题，子问题与原问题是相似的且规模比原问题小，不断拆分后会有一个尽头，即递归边界条件。

  • 子问题：选取l1和l2中更小的的结点，更小结点的next指向合并剩余结点的结果，即$list->next=mergeTwoLists(list->next,list2)$。
  • 递归边界条件： 两个链表中有一个为$\varnothing$

• 归：两个链表中第一个更小的元素。

code

/**
 * 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) return list2;
        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(m+n)$，其中m为l1的链表长度，n为l2的链表长度。

• 空间复杂度:
  $O(m+n)$，递归调用 mergeTwoLists 函数时需要消耗栈空间，栈空间的大小取决于递归调用的深度。

方法二：迭代

思路

我们通过遍历两个链表并生成一个新的链表来终返回，设置一个哑结点dummy，最终返回结果$dummy->next$。
同时设置cur初始值为dummy，来推动新链表的前进。根据题意，取val小的结点，同时让当前结点和新链表的结点向前推动。
最后，会存在有一个链表未遍历完的情况，将剩余的非空链表结在当前结点后面即可。

Code

/**
 * 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) return list2;
        if(list2 == nullptr) return list1;
        ListNode* dummy = new ListNode(), *cur = dummy;
        while(list1 && list2){
            if(list1->val < list2->val){
                cur->next = list1;
                list1 = list1->next;
            }else{
                cur->next = list2;
                list2 = list2->next;
            }
            cur = cur->next;
        }
        cur->next = list1 ? list1 : list2;
        return dummy->next;
    }
};

复杂度

• 时间复杂度:
  $O(m+n)$，其中m为l1的链表长度，n为l2的链表长度。

• 空间复杂度:
  $O(1)$