力扣19. 删除链表的倒数第 N 个结点

19. 删除链表的倒数第 N 个结点

思路

题目要求只遍历一次链表，就不能先计算链表长度，再按顺序遍历去删除了。
而说到倒数第一个想到的就是递归，递归地遍历链表，在末尾返回时每次将n递减。
例如要删除倒数第二个结点，在末尾nullptr返回时n变为1，表示排除最尾结点，此时应该删除的是倒数第1个结点。
当n == 0的时候此时的结点即为待删除的结点。

代码

class Solution {
private:
    // 用第二个参数int n和返回值表示现在要删除的是倒数第几个结点
    int recurion(ListNode* node, int n) {
        // 到末尾时返回，并且从倒数第n个变为倒数第n-1个
        if (node == nullptr)
            return --n;
        n = recurion(node->next, n);
        // 当n==0时表示当前结点就是要删除的前一个结点
        // 之所以是前一个，是因为主函数里传参为n+1
        if (n == 0) {
            ListNode* temp = node->next;
            node->next = temp->next;
            delete temp;
        }
        // 还没到要删除的结点，继续递减n并返回递归
        return --n;
    }
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode* newhead = new ListNode(0, head);  // 新头结点
        recurion(newhead, n + 1);   // 要删除倒数第n个结点，需要在其前面一个结点停下
        return newhead->next;
    }
};