LeetCode(python)——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

思路

快慢指针

（1）快指针先走n步

（2）快慢指针一起走，直到快指针走到链表尾

（3）此时慢指针的下一个节点就是要删除的节点，直接删除即可

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeNthFromEnd(self, head: Optional[ListNode], n: int) -> Optional[ListNode]:
        dummy = ListNode(-1)   # 虚拟节点
        dummy.next = head
        fast, slow = dummy, dummy   # 从虚拟节点开始

        for i in range(n):    # 快指针走n步
            fast = fast.next
        while fast.next:    # 当快指针不是链表尾，快慢指针一起走
            slow = slow.next
            fast = fast.next
        slow.next = slow.next.next    # slow后面的指针是需要删除的节点，删掉即可

        return dummy.next