题目链接
https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/
题目
给你一个链表,删除链表的倒数第
n
个结点,并且返回链表的头结点。进阶:你能尝试使用一趟扫描实现吗?
示例
示例 1:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]示例 2:
输入:head = [1], n = 1
输出:[]示例 3:
输入:head = [1,2], n = 1
输出:[1]
提示
- 链表中结点的数目为
sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz
思路
设置两个指针 fast 和 slow 同时对链表进行遍历,初始时fast在slow前n个位置。则当 fast 遍历到链表的末尾时,slow就恰好处于倒数第 n个节点。
为了便于删除,真实代码编写时我们让slow停在倒数第n个结点的前一结点更方便,因此实际实现时再设置一个虚拟头结点,slow起始位置在这个虚拟头结点的位置,fast位置在虚拟头结点的下一位置(真实头结点),让fast先往前走n步,再两个结点同时走,当fast遍历到链表的末尾时,slow的下一个节点就是我们需要删除的节点。
C++ Code
/**
* 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* vir=new ListNode(0,head);
ListNode* fast=head;
ListNode* slow=vir;
while(n--) fast=fast->next;
while(fast)
{
fast=fast->next;
slow=slow->next;
}
slow->next=slow->next->next;
return vir->next;
}
};