&LeetCode0019& 删除链表的倒数第N个节点

题目

给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后，链表变为 1->2->3->5.

说明：
给定的 n 保证是有效的。

进阶：
你能尝试使用一趟扫描实现吗？

来源：力扣（LeetCode）

思路

首先,要考虑的是，如何找到倒数第N个节点，因为只允许一次遍历，所以不能用一次完整的遍历来统计链表中元素的个数，而是遍历到对应位置就应该移除了；
然后，需要用两个指针，pre 和 cur 指针，先是 cur 指针先向前走N步，如果此时 cur 指向空，说明N为链表的长度，则需要移除的为首元素，那么此时返回 head->next 即可，如果 cur 存在，再继续往下走，此时 pre 指针也跟着走，直到 cur 为最后一个元素时停止，此时 pre 指向要移除元素的前一个元素；
最后，再修改指针跳过需要移除的元素即可，

C++代码

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) 
    {
        if (!head -> next)
            return NULL;
        
        ListNode *pre = head, *cur = head;

        for (int i = 0; i < n; ++i)
        {
            cur = cur -> next;
        }

        if(!cur)
        {
            return head -> next;
        }

        while (cur -> next)
        {
            cur = cur -> next;
            pre = pre -> next;
        }
        pre -> next = pre -> next -> next;
        return head;
    }
};