[数据结构与算法]LeetCode(Python实现)—删除链表的倒数第 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 个结点，并且返回链表的头结点。

解题思路

这道题相对比较简单，先获取链表的总长度，然后循环到要删除的结点的前一个结点进行删除节点操作。

代码

通过计算链表长度来对节点进行删除

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        def length(head: ListNode):
            # 链表长度
            # 初始指针指向head
            cur = head
            count = 0
            # 指针指向None 表示到达尾部
            while cur is not None:
                count += 1
                # 指针下移
                cur = cur.next
            return count
        # 建立头节点
        dummy = ListNode(0, head)
        # 将头节点赋值给枚举节点
        cur = dummy
        for i in range(length(head) - n):
            # 节点移动到指定位置
            cur = cur.next
        # 删除节点
        cur.next = cur.next.next
        # 返回头节点后的数据
        return dummy.next