19. 删除链表的倒数第 N 个结点
给你一个链表,删除链表的倒数第 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
解答思路:
- 先统计链表长度ListSize
- 然后再找到ListSize-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) {
int ListSize = 0;
ListNode *p = head;
while(p){
p = p->next;
ListSize++; //统计链表长度
}
delete(p); //当链表长度 ListSize < 2 时,也就是只有一个结点。
if(ListSize < 2){
return NULL;
}
if(ListSize == n){
return head->next; //当链表长度 ListSize == n 时,也就是删除第一个结点。
}
ListNode * ans = head;
for(int i = 1;i < ListSize-n;i++){//找到第长度减n位置的结点
ans = ans->next;
}
ans->next = ans->next->next;// 将倒数第n个位置删除
return head;//返回头结点
}
};