思路
题目要求只遍历一次链表,就不能先计算链表长度,再按顺序遍历去删除了。
而说到倒数第一个想到的就是递归,递归地遍历链表,在末尾返回时每次将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;
}
};