给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
'''
思路:双指针法
设定两个指针pre和post,pre指向head, post指向后面的节点,与pre间隔n个位置,
然后两个指针同时向后移动,直到post.next=None,此时pre刚好指向倒数第n个节点,
的前一个位置
'''
pre = head
post = head
while n:
post = post.next # post与pre间隔n个位置
if not post: # post为None,说明要删除的是正向第一个节点
return head.next
n -= 1
while post.next: # 当post.next != None时,两个指针同时向后移动
pre = pre.next
post = post.next
# 此时post指向被删除节点的前一个位置
pre.next = pre.next.next # 删除倒数第n个节点
return head