LeetCode 21. 合并两个有序链表

文章目录

• 题目
• 思路
• 代码

题目

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 均按 非递减顺序 排列

思路

算法：
双指针算法 O(m+n)

1. l1表示当前链表1枚举到的位置，l2表示当前链表2枚举到的位置，dummy表示新开存结果的链表
2. 当l1和l2都不为空
   如果l1.val <= l2.val，则把l1.val加入到dummy链表的后面
   如果l1.val > l2.val，则把l2.val加入到dummy链表的后面
3. 当最后l1或者l2存在一个为空时，只有把存在剩余元素的链表继续拼接到dummy后面即可

时间复杂度：n和m分别表示两个链表的长度，所以总的时间复杂度为 O(m+n) .

代码

C++代码：

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

python3代码：

class Solution:
    def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
        # iterative
        dummy = cur = 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 or l2
        return dummy.next