给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
思路:
就和剑指offer上那道返回倒数第k个节点思路一样,不过这道题的返回变成了删除,所以更复杂一点
创建两个指针pq,p先走,隔n+1个节点之后q再走,之所以是n+1,是因为删除第n个节点是要n-1节点的next指向n的next
即:
q->next = q->next->next;
来执行删除
但这里会产生一些问题,比如当链表长度不足2时会产生空指针,或链表长度等于n时答案错误,如[1,2]求倒数第二个数就会输出错误答案1
解决方法是在头结点前插入一个节点,让pq初始指向这个节点
ac代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode newHead(0);
newHead.next = head;
ListNode* pre = &newHead;
ListNode* p = &newHead;
ListNode* q = &newHead;
int count = 0;
while(p){
p = p->next;
if(count > n) q = q->next;
count++;
}
q->next = q->next->next;
return pre->next;
}
};