19. 删除链表的倒数第N个节点
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2. 当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
解题思路
遍历每个结点存入栈中,最后pop出n个结点,并修改连接关系。这里设置哑结点防止头部结点被删除。
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
dummyHead = ListNode(next=head)
node_stack = [dummyHead]
while head:
node_stack.append(head)
head = head.next
for i in range(n):
node = node_stack.pop(-1)
node_last = node_stack.pop(-1)
node_last.next = node.next
return dummyHead.next