✨题目
给定一个链表,删除链表的倒数第n个节点
,并且返回链表的头结点
✨样例
输入:head = [1,2,3,4,5], n = 2 输出:[1,2,3,5]
输入:head = [1], n = 1 输出:[]
✨解题
由于是单链表,造成删除倒数第 n 个节点不好做,所以不妨转换成删除正数的节点,用两个指针去遍历,pre指针总是记录当前节点的前一个节点,最后只需要对删除头节点的情况进行特殊处理一下即可。
/**
* 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 == NULL && n == 1){
return NULL;
}
ListNode* node = head;
int sum = 0;
while(node != NULL){
sum ++;
node = node->next;
}
node = head;
ListNode* preNode = head;
int m = sum - n;
while(m > 0){
preNode = node;
node = node->next;
m--;
}
preNode->next = node->next;
if(preNode == head && preNode == node){
return head->next;
}
delete node;
return head;
}
};