链接
https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/
耗时
解题:8 min
题解:14 min
题意
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
思路
要删除倒数第 n 个节点,需要先找到它前面的那个节点,即倒数第 n+1 个节点,快慢指针可以做到。但需要注意的是,倒数第 n 个节点可能是头节点,它没有前一个节点,需要特殊判断一下。
先将快指针移动 n 次,到达第 n 个节点后面,如果指向空,说明链表长度就是 n,要删除的是头节点,直接返回头节点的 next 即可;如果不为空,则将快指针再向后移动一位,此时快慢指针相距 n+1 位,接下来使得快慢指针同时移动,直到快指针指向空,此时慢指针指向倒数第 n+1 个节点,最后将慢指针指向节点的 next 直接指向下下一个节点,完成删除倒数第 n 个节点。
时间复杂度: O ( l e n g t h ) O(length) O(length)
AC代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* fast = head;
ListNode* slow = head;
for(int i = 0; i < n; ++i) fast = fast->next;
if(fast == NULL) return head->next;
fast = fast->next;
while(fast != NULL) {
fast = fast->next;
slow = slow->next;
}
slow->next = slow->next->next;
return head;
}
};