原题链接:Remove Nth Node From End of List
题目内容:
Given a linked list, remove the n-th node from the end of list and return its head.
Example:
Given linked list: 1->2->3->4->5, and n = 2.
After removing the second node from the end, the linked list becomes 1->2->3->5.
Note:
Given n will always be valid.
Follow up:
Could you do this in one pass?
题目翻译:
给定链表,要求一趟遍历删除倒数第n个元素。
Tips:设置前置指针标记结尾,与目标指针相差n位。
Python
# 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
"""
if not head.next:
return None
foe = head
current = head
for _ in range(n):
foe = foe.next
if not foe:
return head.next
while foe.next:
foe = foe.next
current = current.next
current.next = current.next.next
return head
Python2
class Solution:
def removeNthFromEnd(self, head, n):
def index(node):
if not node:
return 0
i = index(node.next) + 1
if i > n:
node.next.val = node.val
return i
index(head)
return head.next