来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
给你一个链表,删除链表的倒数第 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
结题思路:
首先需要先定位到倒数第 N 个节点,但是该节点是需要删除的,所以我们应该是定位到倒数第 N + 1 个节点,然后将该节点的 Next 指针指向 Next.Next 元素即可。
1、定位到倒数第 N + 1 个节点,可以先将所有元素均压进一个队列里,然后算出总长,再取倒数第 N + 1 个节点。
2、也可以将所有节点都入栈,然后取出N + 1 个节点,然后操作出来的最有一个节点。
3、使用双指针方式,一个先走 N 步,然后再同时往下遍历,直到快指针先遍历完,当前慢指针的位置就是倒数第 N 个节点,但是不是很好操作修改,所以可以在头节点创建一个虚拟节点,虚拟节点的 Next 指针指向头节点,然后慢指针从当前虚拟节点出发即可实现当快指针从头结点出发并先走 N 步,最后再与慢指针一同出发,直到快指针到达结尾时,慢指针所在位置正好是倒数第 N + 1 个节点,然后将该节点的 Next 指针指向 Next.Next 元素即可。
双指针 go 代码示例:
func removeNthFromEnd(head *ListNode, n int) *ListNode {
_head := &ListNode{Next: head}
fast, slow := head, _head
// 快指针先走 N 步
for i := 0; i < n; i ++ {
fast = fast.Next
}
// 快慢指针一起往后遍历,直到快指针遍历完毕
for ; fast != nil; fast = fast.Next {
slow = slow.Next
}
// 删除倒数第 N 个元素即可
slow.Next = slow.Next.Next
return _head.Next
}