21删除链表的倒数第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]
提示:
链表中结点的数目为 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 {
dummy := &ListNode{Next:head}
fast := dummy
// # 1
// s f
// * 1,2,3,4,5
for n > 0 && fast.Next != nil {
fast = fast.Next
n --
}
// # 2
// s f
// * 1,2,3,4,5
if n == 0 {
slow := dummy
for fast != nil && fast.Next != nil {
fast = fast.Next
slow = slow.Next
}
tmp := slow.Next
if tmp != nil {
slow.Next = tmp.Next
tmp.Next = nil
}
}
return dummy.Next
}