链接: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
进阶:你能尝试使用一趟扫描实现吗?
果果念
快慢指针,应该比较经典了吧。
M1:这里使用的时候是直接找倒数第n个,如果还没有找到的话,肯定是删除第一个元素了。
M2:也可以在上面判断的时候,先找倒数第n+1个节点,然后再赋值。
只遍历了一遍。
如有不当之处,欢迎交流~
代码
/**
* 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* ans=head;
ListNode* ansHead=ans;
int num=0;//记录ans和head的下标差值
while(head->next!=nullptr){
if(num==n){
ans=ans->next;
}else num++;
head=head->next;
}
//倒数第sz个,即删除第一个元素
//也可以在上面判断的时候,先找倒数第n+1个节点,然后再赋值。这里使用的时候是直接找倒数第n个,如果还没有找到的话,肯定是删除第一个元素了。
if(num<n){
return ansHead->next;
}
//删除倒数第n个
ans->next=(ans->next)->next;
return ansHead;
}
};