（python刷题）leetcode 第19题：删除链表的倒数第N个节点

题目在leetcode上的链接为：
https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/

题目描述

解题思路
要实现一遍扫描解题，可以使用快慢指针的思想。要删除倒数第 n 个节点，那么需要找到倒数第 n+1 个节点才行。先初始化两个指针 p1=p2=head，然后让快指针 p1 先往前走 n 步，然后让 p2，p1 同时走，直到 p1.next 为空时，p2 就为倒数第 n+1 个节点了，此时就可以删除倒数第 n 个节点。
上述算法中需要注意一个特殊情况，当 n 刚好等于链表长度时，并不存在倒数第 n 个节点，那么对于这一种情况，需要在 p1 先往前走 n 步后，判断 p1 是否已经为空，如果此时 p1 为空的话，就说明 n 刚好等于链表长度，此时删除的倒数第 n 个节点即为头节点，那么只需要删除头节点后直接返回新的头节点即可。

下面以题目中的示例进行算法的演示，给定的链表为 1->2->3->4->5，n=2
1）初始化 p1=p2=head，此时 p1,p2都在 1 的节点处
2）让 p1 先走 n=2 步，此时 p1 走到了 3 的节点处，p2 在 1 的节点处
3）判断 p1 是否为空，此时不为空，因此存在倒数第 n+1 个节点
4）同时让 p2，p1走，直到 p1.next 为空时停止，此时 p1 在 5 的节点处，p2 在 3 的节点处（即倒数第 n+1=3 个节点）
5）删除倒数第 n 个节点

python代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        if head == None:
            return head
        p1 = p2 = head
        # 先让p1走n步
        while n > 0:
            p1 = p1.next
            n -= 1
        # n刚好等于链表的长度，则需要删除头结点
        if p1 == None:
            newhead = head.next
            head.next = None
            return newhead
        # p1,p2同时走，直到p1.next为空，此时p2为倒数第n+1个节点
        while p1.next != None:
            p1 = p1.next
            p2 = p2.next
        # 删除倒数第n个节点
        temp = p2.next
        p2.next = p2.next.next
        temp.next = None

        return head