leetcode No.19 删除链表的倒数第N个节点 (python3实现)

来源

https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/

题目描述

给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.

当删除了倒数第二个节点后，链表变为 1->2->3->5.
说明：

给定的 n 保证是有效的。

进阶：

你能尝试使用一趟扫描实现吗？

代码实现

# 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:
        # 增加一个特殊节点方便边界判断
        p = ListNode(-1)
        p.next, a, b = head, p, p
        while n > 0 and b:
            b, n = b.next, n-1
        if not b:
            return head
        while b.next:
            a, b = a.next, b.next
        a.next = a.next.next
        return p.next