LeetCode19. 删除链表的倒数第N个节点 Python3

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Victordas/article/details/82883781

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

示例：

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

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

给定的 n 保证是有效的。

进阶：

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

这题要求时间复杂度是O（n）

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

class Solution:
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        if head.next == None:
            return []
        else:
            pre = head
            end = head
            for i in range(n):
                end = end.next
            if end == None:
                return head.next
            while end.next:
                end = end.next
                pre = pre.next
            pre.next = pre.next.next
            return head