题目描述
给你一个链表,删除链表的倒数第 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