原题目链接:Link。
拿到题目,其实有一个简单的 idea:
- 遍历列表,求出列表长度 \(len\);
- 从 \(head\) 遍历列表到第 \(len - n\) 个;
- 普通的删除操作。
容易写出代码:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
int len = 1;
ListNode* cur = head;
while (cur -> next != nullptr)
cur = cur -> next, len++;
cur = head;
for (int i = 1; i < len - n; i++) cur = cur -> next;
cur -> next = cur -> next -> next;
return head;
}
};
然后提交,你会发现第一个点都过不了。(哈哈哈哈。)观察样例:
[1]
1
会发现 \(cur\) 就是 \(head\),然后 head -> next
是 nullptr,然后再来一个 -> next
……RE 快乐。
所以就产生了一种技巧,叫做哑节点,其 \(next\) 指向 \(head\),这样我们就不用判断边界了。
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy = new ListNode(0, head);
ListNode* cur = head;
int len = 1;
for (; cur -> next != nullptr; cur = cur -> next)
len++; // 得到链表长度
cur = dummy; // 从哑节点开始遍历
for (int i = 1; i <= len - n; i++) cur = cur -> next;
cur -> next = cur -> next -> next;
return dummy -> next;
// 如果要求释放空间,就可以先用一个 ListNode* 存储 dummy -> next,然后 delete dummy
}
};