题目描述
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
算法思想
采用一次扫描的办法,需要定位到删除节点的前面的节点
因此指针p往后挪动的个数为n+1次
注意如果p挪动之后,碰到空,说明删除的是头结点
使用两个快慢指针,目的是定位到倒数第n个节点。一个从头还是挪动,一个从p开始挪动,知道p为空,就找到了前面的指针,之后直接删除即可
通过代码
class Solution:
def removeNthFromEnd(self, head, n):
"""
:type head: ListNode
:type n: int
:rtype: ListNode
"""
if head == None or n < 1:
return None
p = head
for i in range(n):
p = p.next
#如果p为None,那么需要删除头结点
if p == None:
tempHead = head
head = head.next
del tempHead
return head
preNode = head
while p.next:
preNode = preNode.next
p = p.next
deleteNode = preNode.next
preNode.next = deleteNode.next
del deleteNode
return head
运行结果