题目在leetcode上的链接为:
https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/
题目描述
解题思路
要实现一遍扫描解题,可以使用快慢指针的思想。要删除倒数第 n 个节点,那么需要找到倒数第 n+1 个节点才行。先初始化两个指针 p1=p2=head,然后让快指针 p1 先往前走 n 步,然后让 p2,p1 同时走,直到 p1.next 为空时,p2 就为倒数第 n+1 个节点了,此时就可以删除倒数第 n 个节点。
上述算法中需要注意一个特殊情况,当 n 刚好等于链表长度时,并不存在倒数第 n 个节点,那么对于这一种情况,需要在 p1 先往前走 n 步后,判断 p1 是否已经为空,如果此时 p1 为空的话,就说明 n 刚好等于链表长度,此时删除的倒数第 n 个节点即为头节点,那么只需要删除头节点后直接返回新的头节点即可。
下面以题目中的示例进行算法的演示,给定的链表为 1->2->3->4->5,n=2
1)初始化 p1=p2=head,此时 p1,p2都在 1 的节点处
2)让 p1 先走 n=2 步,此时 p1 走到了 3 的节点处,p2 在 1 的节点处
3)判断 p1 是否为空,此时不为空,因此存在倒数第 n+1 个节点
4)同时让 p2,p1走,直到 p1.next 为空时停止,此时 p1 在 5 的节点处,p2 在 3 的节点处(即倒数第 n+1=3 个节点)
5)删除倒数第 n 个节点
python代码
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
if head == None:
return head
p1 = p2 = head
# 先让p1走n步
while n > 0:
p1 = p1.next
n -= 1
# n刚好等于链表的长度,则需要删除头结点
if p1 == None:
newhead = head.next
head.next = None
return newhead
# p1,p2同时走,直到p1.next为空,此时p2为倒数第n+1个节点
while p1.next != None:
p1 = p1.next
p2 = p2.next
# 删除倒数第n个节点
temp = p2.next
p2.next = p2.next.next
temp.next = None
return head