给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
示例 2:
输入:head = [1], n = 1
输出:[]
示例 3:
输入:head = [1,2], n = 1
输出:[1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
用两个指针,一个快一个慢,当快的移动了n-1个的时候,慢的开始走,当快的到结束的时候,慢的到达第n个位置的前一个位置,将它的指针指向要删除的下一个结点即可。
/**
* 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) {
if(head->next == nullptr || n == 0) return nullptr;
ListNode *heada=head;
ListNode *headb=nullptr;
for(int i=0; i<n; i++)
{
heada = heada->next;
}
if(heada == nullptr)
return head->next;
headb = head;
while(heada->next != nullptr)
{
heada = heada->next;
headb = headb->next;
}
headb->next = headb->next->next;
return head;
}
};