删除链表中倒数第n个节点

remove Nth node from end of list

描述：

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

样例：

　　给出链表1->2->3->4->5->null和 n = 2.

　　删除倒数第二个节点之后，这个链表将变成1->2->3->5->null.

挑战：

　　O(n)时间复杂度

思路：

　　先求得链表长度，查询链表依次时间O(n)，第二次找到倒数n位置，删除之，时间共O(n)+O(n)~ O(n)

"""
Definition of ListNode
class ListNode(object):

    def __init__(self, val, next=None):
        self.val = val
        self.next = next
"""
class Solution:
    """
    @param head: The first node of linked list.
    @param n: An integer.
    @return: The head of linked list.
    """
    def removeNthFromEnd(self, head, n):
        # write your code here
        dummy = ListNode(0)
        dummy.next = head
        
        # for length
        len = 0
        pre = dummy
        while pre.next:
            len += 1
            pre = pre.next
            
        # find the Nth node from end of list
        cnt = 0 
        tarCnt = len - n
        pre = dummy
        while pre.next:
            if cnt == tarCnt:
                pre.next = pre.next.next
                break
            else:
                cnt +=1
                pre = pre.next
                
        return dummy.next