这题如果可以遍历两次链表,就很简单:先统计链表长度,第二遍的时候在进行删除操作。如果只遍历一次,我采用的是快慢指针法。
【题目描述】
【示例】
示例 2:
输入:head = [1], n = 1
输出:[]
示例 3:
输入:head = [1,2], n = 1
输出:[1]
提示:
链表中结点的数目为 sz
1 ≤ s z ≤ 30 1 \le sz \le 30 1≤sz≤30
0 ≤ N o d e . v a l ≤ 100 0 \le Node.val \le 100 0≤Node.val≤100
1 ≤ n ≤ s z 1 \le n \le sz 1≤n≤sz
【解题过程】
【思路】
想用快慢指针法,前一个指针先走N步,然后两个指针一起移动,当先走的指针到达末尾,后一指针刚好在倒数第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==1){
return nullptr;
}
ListNode* right=head;
ListNode* left=head;
while(n--){
right=right->next;
}
while(right!=nullptr){
left=left->next;
right=right->next;
}
if(left->next!=nullptr){
left->next=left->next->next;
}
return head;
}
};
运行结果:
样例都跑不出来,删错了元素。。。。。
想了很久没想出来哪里不对劲,后来去看了官解代码:
官解链接
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy = new ListNode(0, head);
ListNode* first = head;
ListNode* second = dummy;
for (int i = 0; i < n; ++i) {
first = first->next;
}
while (first) {
first = first->next;
second = second->next;
}
second->next = second->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
};
它设置了一个空的头结点,用于初始化second,这样就产生了一个结点的时间差,也不会出现first=null时,还在执行first=first->next的情况。
仔细看了官解的图,才明白这个逻辑关系,看来做题还是要画画图更清晰。