跳至主要內容

剑指 Offer II 021. 删除链表的倒数第 n 个结点

ITCharge大约 1 分钟

剑指 Offer II 021. 删除链表的倒数第 n 个结点open in new window

  • 标签:链表、双指针
  • 难度:中等

题目链接

题目大意

给你一个链表的头节点 head 和一个整数 n

要求:删除链表的倒数第 n 个节点,并且返回链表的头节点。并且要求使用一次遍历实现。

解题思路

常规思路是遍历一遍链表,求出链表长度,再遍历一遍到对应位置,删除该位置上的节点。

如果用一次遍历实现的话,可以使用快慢指针。让快指针先走 n 步,然后快慢指针、慢指针再同时走,每次一步,这样等快指针遍历到链表尾部的时候,慢指针就刚好遍历到了倒数第 n 个节点位置。将该位置上的节点删除即可。

需要注意的是要删除的节点可能包含了头节点。我们可以考虑在遍历之前,新建一个头节点,让其指向原来的头节点。这样,最终如果删除的是头节点,则删除原头节点即可。返回结果的时候,可以直接返回新建头节点的下一位节点。

代码

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        newHead = ListNode(0, head)
        fast = head
        slow = newHead
        while n:
            fast = fast.next
            n -= 1
        while fast:
            fast = fast.next
            slow = slow.next
        slow.next = slow.next.next
        return newHead.next