删除链表的倒数第N个节点,上数据结构课的时候老师讲了一个比较巧妙的方法,快慢指针,即定义快慢两个指针,快指针先走N步(N为要删除的倒数第N个元素的N),然后快慢指针一起走,他两之间的距离始终保持N,这样的话当快指针走到末尾的时候,慢指针所在的位置即为要删除的节点的位置,很巧妙吧?如果没看懂,没关系,简单说,这个N就相当于一把尺子,尺子的两端一个快指针一个慢指针
好了,说完了,看看代码(python3)
# 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:
#删除倒数第N个节点,即可以定义快慢两个指针,快指针先走N步,慢指针再开始走
#这样等到快指针走到最后一个节点时慢指针所在的位置即为要删除的倒数第N个节点
slow = fast = head
for i in range(n):
fast = fast.next
if fast == None:
return haed.next
while fast.next != None:
slow = slow.next
fast = fast.next
slow.next = slow.next.next
return head