解题思路:
根据题目描述, 链表 l1
, l2
是 递增 的,因此容易想到使用双指针 l1
和 l2
遍历两链表,根据 l1.val
和 l2.val
的大小关系确定节点添加顺序,两节点指针交替前进,直至遍历完毕。
引入伪头节点: 由于初始状态合并链表中无节点,因此循环第一轮时无法将节点添加到合并链表中。解决方案:初始化一个辅助节点 dum
作为合并链表的伪头节点,将各节点添加至 dum
之后。
算法流程:
- 初始化: 伪头节点
dum
,节点cur
指向dum
。 - 循环合并: 当
l1
或l2
为空时跳出;- 当
l1.val < l2.val
时:cur
的后继节点指定为l1
,并l1
向前走一步; - 当
l1.val >= l2.val
时:cur
的后继节点指定为l2
,并l2
向前走一步 ; - 节点
cur
向前走一步,即cur = cur.next
。
- 当
- 合并剩余尾部: 跳出时有两种情况,即
l1
为空 或l2
为空。- 若
l1 != null
: 将l1
添加至节点cur
之后; - 否则: 将
l2
添加至节点cur
之后。
- 若
- 返回值: 合并链表在伪头节点
dum
之后,因此返回dum.next
即可。
<,,,,,,,,,,,,,,>
代码:
Python 三元表达式写法 A if x else B
,代表当 x = True
时执行 A
,否则执行 B
。
Python
class Solution:
def trainningPlan(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 trainningPlan(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* trainningPlan(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;
}
};
复杂度分析:
- 时间复杂度 $O(M+N)$ : $M$ , $N$ 分别为链表
l1
,l2
的长度,合并操作需遍历两链表。 - 空间复杂度 $O(1)$ : 节点引用
dum
,cur
使用常数大小的额外空间。