题目描述:
给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
样例:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
算法思路:
正着遍历链表,得到链表全长k,那么要删除的节点正着数就是第 k - n 个,再遍历链表,找到要删除的链表节点,将其删除即可。可以使用虚拟头节点简化代码中对头节点的判断。
时间复杂度:
O(n)
代码实现:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
// 创建虚拟头节点dummy
ListNode* dummy = new ListNode(-1);
dummy->next = head;
// 遍历链表,得到链表总长度 k
int k = 0;
for(auto p = dummy; p; p = p->next) k++;
// 遍历到要删除的链表的前一个节点,所以这里是 k - n - 1
auto p = dummy;
for(int i = 0; i < k - n - 1; i++) p = p->next;
// 删除目标节点。
p->next = p->next->next;
return dummy->next;
}
};