剑指 Offer 25. 合并两个排序的链表
输入两个递增排序的链表,合并这两个链表并使新链表中的节点仍然是递增排序的。
示例1:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4
限制:
0 <= 链表长度 <= 1000
注意:本题与主站 21 题相同:https://leetcode-cn.com/problems/merge-two-sorted-lists/
方法1:递归
递归合并两条链表,每次递归返回合并完成的链表头。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
if not l1:
return l2
if not l2:
return l1
if l1.val < l2.val:
l1.next = self.mergeTwoLists(l1.next, l2)
return l1
else:
l2.next = self.mergeTwoLists(l1, l2.next)
return l2
复杂度分析
- 时间复杂度:O(N) 。其中 N 为较长链表的长度。
- 空间复杂度:O(N) 。递归所需栈空间大小为 N 。
方法2:迭代法
(1) 初始化: 伪头节点 res ,节点 p 指向 res 。
(2) 循环合并: 当
l
1
l_1
l1或
l
2
l_2
l2为空时跳出;
- 当 l 1 . v a l < l 2 . v a l l_1.val < l_2.val l1.val<l2.val 时: p的后继节点指定为 l 1 l_1 l1,并 l 1 l_1 l1向前走一步;
- 当 l 1 . v a l ≥ l 2 . v a l l_1.val≥l_2.val l1.val≥l2.val时:p的后继节点指定为 l 2 l_2 l2 ,并 l 2 l_2 l2向前走一步 ;
- 节点 p 向前走一步,即 p = p.next 。
(3) 合并剩余尾部: 跳出时有两种情况,即 l 1 l_1 l1为空 或 l 2 l_2 l2 为空。
- 若 l 1 ≠ n u l l l_1 \ne null l1=null : 将 l 1 l_1 l1添加至节点 p 之后;
- 否则: 将 l 2 l_2 l2添加至节点 p 之后。
(4) 返回值: 合并链表在伪头节点 res 之后,因此返回 res.next 即可。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
res = p = ListNode(0)
while l1 and l2:
if l1.val < l2.val:
p.next, l1 = l1, l1.next
else:
p.next, l2 = l2, l2.next
p = p.next
p.next = l1 if l1 else l2
return res.next