算法---删除链表倒数第n个元素

给定一个长度大于等于n的链表，删除其倒数第n个元素

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
__title__ = ''
__author__ = 'thinkreed'
__mtime__ = '2017/3/21'

"""


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

class Solution(object):
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        thanks to https://discuss.leetcode.com/topic/14692/3-short-python-solutions
        """

        #一个快，一个慢
        fast = slow = head
        #先让快的比慢的先走n个
        for _ in range(n):
            fast = fast.next
        #整个链表长度等于n，所以倒数第n个就是头部
        if not fast:
            return head.next
        #当循环结束时，慢指针slow恰好走到倒数第n个
        while fast.next:
            fast = fast.next
            slow = slow.next
        #删除倒数第n个
        slow.next = slow.next.next

        return head