好处:
使用了虚拟头结点的技巧,也是为了防止出现空指针的情况。
例如:链表 head = [1,2,3,4,5],需要删除倒数第 5 个结点,也就是第一个节点。按照算法逻辑,应该首先找到倒数第 6 个节点。但由于头节点不存在前驱节点,因此我们需要在删除头节点时进行特殊判断。
但是添加一个哑节点(dummy node),让它的next 指针指向链表的头节点。这样一来,头节点的前驱节点就是哑节点本身。
设置虚拟头节点:
dummy = ListNode(0,head)
案例
给你一个链表,删除链表的倒数第 n
个结点,并且返回链表的头结点。
输入:head = [1,2,3,4,5], n = 5 输出:[2,3,4,5]
思路与算法
一种容易想到的方法是,我们首先从头节点开始对链表进行一次遍历,得到链表的长度 L。为了方便删除操作,添加一个哑结点dummy,我们可以从哑节点开始遍历,当遍历到第 L−n+1 个节点时,它的下一个节点就是我们需要删除的节点,这样我们只需要修改一次指针,就能完成删除操作。
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
def getLength(head: ListNode) -> int:
length = 0
while head:
length += 1
head = head.next
return length
# 添加哑结点dummy
dummy = ListNode(0, head)
length = getLength(head)
cur = dummy
for i in range(1, length - n + 1):
cur = cur.next
cur.next = cur.next.next
return dummy.next