-
Notifications
You must be signed in to change notification settings - Fork 0
/
合并两个有序链表.py
31 lines (26 loc) · 1.42 KB
/
合并两个有序链表.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
"""
为了合并两个升序链表为一个新的升序链表,我们可以使用迭代的方式,
比较两个链表的当前节点,然后将较小的节点添加到新链表中。具体步骤如下:
1 创建一个哑节点(dummy node),用于简化边界情况的处理。
2 创建一个指针 current,初始时指向哑节点。
3 遍历两个链表,直到至少有一个链表被遍历完毕。
4 在每次迭代中,比较两个链表当前节点的值,将较小值的节点添加到 current 的 next。
5 移动 current 指针和较小值节点所在链表的指针。
6 遍历结束后,将未结束的链表剩余部分拼接到新链表末尾。
7 返回哑节点的下一个节点,即合并后链表的头节点。
"""
class Solution:
def mergeTwoLists(self, list1: Optional[ListNode], list2: Optional[ListNode]) -> Optional[ListNode]:
dummy = ListNode(0) # 创建哑节点
current = dummy # 当前节点
while list1 and list2:
if list1.val < list2.val:
current.next = list1
list1 = list1.next
else:
current.next = list2
list2 = list2.next
current = current.next
# 如果其中一个链表已经遍历完,将另一个链表的剩余部分连接到当前链表的末尾
current.next = list1 if list1 else list2
return dummy.next # 返回合并后的链表头节点