剑指 Offer II 021. 删除链表的倒数第 n 个结点
大约 1 分钟
剑指 Offer II 021. 删除链表的倒数第 n 个结点
- 标签:链表、双指针
- 难度:中等
题目链接
题目大意
给你一个链表的头节点 head
和一个整数 n
。
要求:删除链表的倒数第 n
个节点,并且返回链表的头节点。并且要求使用一次遍历实现。
解题思路
常规思路是遍历一遍链表,求出链表长度,再遍历一遍到对应位置,删除该位置上的节点。
如果用一次遍历实现的话,可以使用快慢指针。让快指针先走 n 步,然后快慢指针、慢指针再同时走,每次一步,这样等快指针遍历到链表尾部的时候,慢指针就刚好遍历到了倒数第 n 个节点位置。将该位置上的节点删除即可。
需要注意的是要删除的节点可能包含了头节点。我们可以考虑在遍历之前,新建一个头节点,让其指向原来的头节点。这样,最终如果删除的是头节点,则删除原头节点即可。返回结果的时候,可以直接返回新建头节点的下一位节点。
代码
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
newHead = ListNode(0, head)
fast = head
slow = newHead
while n:
fast = fast.next
n -= 1
while fast:
fast = fast.next
slow = slow.next
slow.next = slow.next.next
return newHead.next