题目描述
给你一个链表,删除链表的倒数第 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]
提示:
- 链表中结点的数目为 sz
- 1 <= sz <= 30
- 0 <= Node.val <= 100
- 1 <= n <= sz
代码及注释
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func removeNthFromEnd(head *ListNode, n int) *ListNode {
// 初始化当前节点为链表头部
cur := head
// 创建一个哑节点作为链表的起始点
dummy := &ListNode{
Next: head,
}
// 初始化前一个节点为哑节点
pre := dummy
// 遍历链表
for cur != nil {
// 如果 n 大于 0,继续前进直到 n 为 0
if n > 0 {
n--
} else {
// 如果 n 小于等于 0,前一个节点也前进
pre = pre.Next
}
cur = cur.Next
}
// 删除倒数第 n 个节点
pre.Next = pre.Next.Next
// 返回链表的头部(哑节点的下一个节点)
return dummy.Next
}
代码解释
这里使用了两个指针,cur
和 pre
,它们都从链表的起始位置开始。首先,cur
指针先移动 n
步,然后 cur
和 pre
指针一起向前移动,直到 cur
到达链表的末尾。此时,pre
指针就指向了需要删除的节点的前一个节点。