LeetCode 19.删除链表的倒数第N个节点
给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
示例1:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
示例 2:
输入:head = [1], n = 1
输出:[]
示例 3:
输入:head = [1,2], n = 1
输出:[1]
提示:
链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz
代码:
class Solution:
def removeNthFromEnd(self, head: Optional[ListNode], n: int) -> Optional[ListNode]:
# 创建虚拟头节点,避免删除头节点时的特殊处理
newHead = ListNode(next=head)
slow = fast = newHead # 初始化双指针
# 让 fast 指针先移动 n + 1 步
for _ in range(n + 1):
fast = fast.next
# 同时移动 slow 和 fast,直到 fast 到达链表末尾,即 fast 指针为 null 时
while fast:
slow = slow.next
fast = fast.next
# 删除倒数第 n 个节点,即现在慢指针所指向的节点的后一个节点
slow.next = slow.next.next
# 返回修改后的链表头节点
return newHead.next