力扣刷题-链表-删除链表的倒数第N个节点

news2024/10/6 8:40:19

19.删除链表的倒数第N个节点

给你一个链表，删除链表的倒数第 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个，从前到后就是需要有个索引，但现在问题在于链表总的长度并不知道，所以需要第一步先去找到链表的长度：通过定义一个size，然后对链表进行遍历，逐渐加1，获取链表长度。然后定义索引为index=size-n。再遍历到索引处，执行删除。

class ListNode(object):
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

class Solution(object):
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        dummy_node = ListNode(next=head)
        cur = dummy_node
        size = 0 # 链表长度
        while cur.next:
            cur = cur.next
            size += 1
        temp = dummy_node
        index = size - n
        for i in range(index):
            temp = temp.next
        temp.next = temp.next.next
        return dummy_node.next

进阶：你能尝试使用一趟扫描实现吗？
可以用快慢指针法。
双指针的经典应用，如果要删除倒数第n个节点，让fast移动n步，然后让fast和slow同时移动，直到fast指向链表末尾。删掉slow所指向的节点就可以了。（比较巧妙）
思路是这样的，但要注意一些细节。
分为以下几步：

定义虚拟节点
定义fast指针和slow指针，初始值为虚拟头结点，如图：

fast首先走n + 1步，为什么是n+1呢，因为只有这样同时移动的时候slow才能指向删除节点的上一个节点（方便做删除操作），如图：

fast和slow同时移动，直到fast指向末尾，如图：

注意是 fast为None 而不是 fast.next 为None

删除slow指向的下一个节点，如图：

class Solution(object):
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        dummy_node = ListNode(next=head) # 定义一个虚拟节点
        fast = dummy_node
        slow = dummy_node
        for i in range(n+1): # fast先走n+1步
            fast = fast.next
        while fast != None: # fast和slow同时走
            fast = fast.next
            slow = slow.next
        slow.next = slow.next.next
        return dummy_node.next