题解 | 删除链表的倒数第n个节点
删除链表的倒数第n个节点
https://www.nowcoder.com/practice/f95dcdafbde44b22a6d741baf71653f6?tpId=295&tqId=727&sourceUrl=%2Fexam%2Foj%3FquestionJobId%3D10%26subTabName%3Donline_coding_page
思路
本质是删除某个结点,根据链表长度进行结点的定位,然后分情况讨论
步骤
- 老样子 if (head == nullptr) return nullptr;
- 设置一个虚拟头结点dummy(不排除要删除的结点在第一个)指向头结点
- 创建临时指针 temp 指向虚拟头结点,设置变量统计链表长度 len
- 创建指针 p 指向虚拟头结点,让 p 前进到要删除结点的前一个位置
- 如果n=1,那直接让 p 的下一个结点指向空即可;如果大于1,则让 p->next=p->next->next
- 此时还有可能 p 为 dummy,所以重新定位 head 为 dummy->next
- 返回 head
正确答案
/**
* struct ListNode {
* int val;
* struct ListNode *next;
* ListNode(int x) : val(x), next(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param head ListNode类
* @param n int整型
* @return ListNode类
*/
ListNode* removeNthFromEnd(ListNode* head, int n) {
// write code here
if (head == nullptr) return nullptr;
auto* dummy = new ListNode(0);
dummy->next = head;
ListNode* p = dummy;
ListNode* temp = dummy;
int len = 0;
while (temp->next != nullptr) {
len++;
temp = temp->next;
}
for (int i = 0; i < len - n; i++) {
p = p->next;
}
if (n == 1) p->next = nullptr;
else p->next = p->next->next;
head = dummy->next;
return head;
}
};