问题描述
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
链表定义:
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
Method 1
我们注意到这个问题可以容易地简化成另一个问题:删除从列表开头数起的第 (L−n+1) 个结点,其中 L 是列表的长度。只要我们找到列表的长度 L,这个问题就很容易解决。
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
if n < 1:
return
count = 0
cur = head
while cur:
count = count + 1
cur = cur.next
loc = count - n
if loc == 0:
head = head.next
return head
pre = head
cur = pre.next
for i in range(loc-1):
pre = pre.next
cur = pre.next
cur = cur.next
pre.next = cur
return head
Method 2
基本思想:d代表删除节点,last为相对于cur节点尾节点的位置,同时移动cur节点和last节点,当节点last到达链表尾部时,删除cur节点后一节点
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
if n < 1:
return
cur = head
last = head
for i in range(n):
last = last.next
if last is None:
head = head.next
return head
while last.next:
cur = cur.next
last = last.next
d = cur.next
cur.next = d.next
return head