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

题目链接

https://leetcode.cn/problems/remove-nth-node-from-end-of-list/

题目描述

给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。

进阶：你能尝试使用一趟扫描实现吗？

思路

双指针的经典应用，如果要删除倒数第n个节点，让fast移动n步，然后让fast和slow同时移动，直到fast指向链表末尾。删掉slow所指向的节点就可以了。

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

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

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

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

代码

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

#暴力解法
class Solution:
    def get_size(self, head: ListNode):       #计算链表长度
        cur = head
        i = 0
        while cur != None:
            i += 1
            cur=cur.next
        return i
    def removeNthFromEnd(self, head:ListNode, n: int) -> ListNode:
        dummy_head=ListNode(next=head)      #虚拟头节点
        size=self.get_size(head)
        cur=dummy_head
        i=0
        while True:
            if i==size-n:
                cur.next=cur.next.next
                return dummy_head.next
            else:
                cur=cur.next
                i+=1



#快慢指针法
# 先让快指针走n步，然后让快慢指针同步，当快指针走完时，满指针指向需要删除节点的上一个节点

class Solution2:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        dummy_head=ListNode(next=head)
        fast=dummy_head
        slow=dummy_head
        i=0
        while fast.next != None:   #此时的fast为上一轮中的fast，fast在最后一轮循环中走到最后一个节点
            if i < n:
                fast=fast.next
            if i>=n:
                fast=fast.next
                slow=slow.next
            i+=1
        slow.next=slow.next.next

        return dummy_head.next



a=ListNode(1)
b=ListNode(2)
c=ListNode(3)
a.next=b
b.next=c
# head=ListNode(next=a)

def print_linklist(lk):  #lk是head
    while lk!=None:
        print(lk.val, end=',')
        lk = lk.next


print_linklist(a)

solu=Solution2()
solu.removeNthFromEnd(a,1)
print_linklist(a)