Robin Harcker

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
}