leetcode 19
链接:力扣 。
题目:
给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
进阶:你能尝试使用一趟扫描实现吗?
示例1:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
示例2:
输入:head = [1], n = 1
输出:[] 示例 3:
示例3:
输入:head = [1,2], n = 1
输出:[1]
思路:
双指针的经典应用,如果要删除倒数第n个节点,让fast与slow快n-1步,然后让fast和slow同时移动,直到fast指向链表末尾。slow所指向的就是需要删除的结点。
但是这样又存在一个问题,我们要在链表中删除一个结点,必须要有指针指向这个结点的前驱,这样又需要设置一个指针。因此我们让fast与slow快n步,这样slow所指的下一个结点即为所需要删除元素。
还有一点需要提及,为了便于统一操作,采用虚拟头结点。
参考代码:
/**
* 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 *vHead = new ListNode(0);
vHead->next = head;
ListNode *slow = vHead, *fast = vHead;
while (n--) {
fast = fast->next;
}
while (fast->next) {
slow = slow->next;
fast = fast->next;
}
ListNode *tmp = slow->next;
slow->next = tmp->next;
delete tmp;
return head;
}
};