剑指 Offer 25. 合并两个排序的链表 - Leetcode

解题思路：

根据题目描述，链表 $l_1$ , $l_2$ 是递增的，因此容易想到使用双指针 $l_1$ 和 $l_2$ 遍历两链表，根据 $l_1.val$ 和 $l_2.val$ 的大小关系确定节点添加顺序，两节点指针交替前进，直至遍历完毕。
引入伪头节点： 由于初始状态合并链表中无节点，因此循环第一轮时无法将节点添加到合并链表中。解决方案：初始化一个辅助节点 $dum$ 作为合并链表的伪头节点，将各节点添加至 $dum$ 之后。

{:width=400}

算法流程：

初始化： 伪头节点 $dum$ ，节点 $cur$ 指向 $dum$ 。
循环合并： 当 $l_1$ 或 $l_2$ 为空时跳出；
1. 当 $l_1.val < l_2.val$ 时： $cur$ 的后继节点指定为 $l_1$ ，并 $l_1$ 向前走一步；
2. 当 $l_1.val \geq l_2.val$ 时： $cur$ 的后继节点指定为 $l_2$ ，并 $l_2$ 向前走一步；
3. 节点 $cur$ 向前走一步，即 $cur = cur.next$ 。
合并剩余尾部： 跳出时有两种情况，即 $l_1$ 为空或 $l_2$ 为空。
1. 若 $l_1 \ne null$ ：将 $l_1$ 添加至节点 $cur$ 之后；
2. 否则：将 $l_2$ 添加至节点 $cur$ 之后。
返回值： 合并链表在伪头节点 $dum$ 之后，因此返回 $dum.next$ 即可。

<,,,,,,,,,,,,,,,>

复杂度分析：

时间复杂度 $O(M+N)$ ： $M, N$ 分别为链表 $l_1$, $l_2$ 的长度，合并操作需遍历两链表。
空间复杂度 $O(1)$ ： 节点引用 $dum$ , $cur$ 使用常数大小的额外空间。

代码：

Python 三元表达式写法 A if x else B ，代表当 $x = True$ 时执行 $A$ ，否则执行 $B$ 。

Python

class Solution:
    def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
        cur = dum = ListNode(0)
        while l1 and l2:
            if l1.val < l2.val:
                cur.next, l1 = l1, l1.next
            else:
                cur.next, l2 = l2, l2.next
            cur = cur.next
        cur.next = l1 if l1 else l2
        return dum.next

Java

class Solution {
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode dum = new ListNode(0), cur = dum;
        while(l1 != null && l2 != null) {
            if(l1.val < l2.val) {
                cur.next = l1;
                l1 = l1.next;
            }
            else {
                cur.next = l2;
                l2 = l2.next;
            }
            cur = cur.next;
        }
        cur.next = l1 != null ? l1 : l2;
        return dum.next;
    }
}

C++

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        ListNode* dum = new ListNode(0);
        ListNode* cur = dum;
        while(l1 != nullptr && l2 != nullptr) {
            if(l1->val < l2->val) {
                cur->next = l1;
                l1 = l1->next;
            }
            else {
                cur->next = l2;
                l2 = l2->next;
            }
            cur = cur->next;
        }
        cur->next = l1 != nullptr ? l1 : l2;
        return dum->next;
    }
};