19、删除链表的倒数第N个结点
文章出现的代码都是python3
题目
给你一个链表,删除链表的倒数第n个结点,并返回链表的头结点
示例1:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
示例2:
输入:head = [1], n = 1
输出:[]
示例3:
输入:head = [1,2], n = 1
输出:[1]
思路
要删除倒数第n个结点,需要找到倒数第n个结点的前驱结点,删除其的下一个结点即可。而链表的长度是未知的,所以第一想法就是先计算链表的长度,然后再找到第n个结点。还有一种方法就是使用快慢指针,这样当快指针走到尾结点的时候,此时只需要满指针刚好走到倒数第n个结点,或者倒数第n个结点的前驱结点即可
解法1
先计算链表长度,然后找到第n个结点的前驱结点,删除即可。
class Solution:
# 获取链表长度
def getLinkedLength(self, root: ListNode):
l, cur = 0, root
while cur:
l += 1
cur = cur.next
return l
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
if not head: return head
pre = self.getLinkedLength(head) - n
# 创建虚拟头结点,处理删除头结点的情况
vHead = ListNode(0, head)
cur = vHead
while pre:
cur = cur.next
pre -= 1
cur.next = cur.next.next
return vHead.next
解法2
使用快慢指针,快指针比慢指针多走n+1个长度,那么当快指针走到尾结点的时候,慢指针刚好走到要删除的前驱结点。
class Solution:
def removeNthFromEnd1(self, head: ListNode, n: int) -> ListNode:
# 创建虚拟头结点,处理删除头结点的情况
vHead, fast = ListNode(0, head), head
# 将慢指针指向虚拟头结点
slow = vHead
while n:
fast = fast.next
n -= 1
while fast:
fast = fast.next
slow = slow.next
slow.next = slow.next.next
return vHead.next